Mlflow project definition (#22)

* Change location of classification_report.json * Use flass command in MLproject
flikka · Jun 18, 2020 · 3849b82 · 3849b82
1 parent 237c5e0
commit 3849b82
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/MLproject b/MLproject
@@ -0,0 +1,11 @@
+name: flass-mlproject
+
+docker_env:
+  image: docker.io/kminaister/flass:latest
+
+entry_points:
+  main:
+    parameters:
+      dataset: string
+      subset: {type: int, default: -1}
+    command: "flass --dataset {dataset} --subset {subset}"
diff --git a/flass/cli.py b/flass/cli.py
@@ -83,7 +83,7 @@ def flass(plot, batch_size, epochs, dataset, model_type, subset, lime):
         report = classification_report(
             y_test, y_predicted, target_names=class_names, output_dict=True
         )
-        classification_report_file = "classification_report.json"
+        classification_report_file = "/tmp/classification_report.json"
         with open(classification_report_file, "w") as file:
             json.dump(report, file)
         logger.info(

diff --git a/kfp/flassline.py b/kfp/flassline.py
@@ -28,7 +28,7 @@ def flass_onestep(
             batch_size,
         ],
         output_artifact_paths={
-            "classification-report-json": "/home/appuser/classification_report.json"
+            "classification-report-json": "/tmp/classification_report.json"
         },
     )