Added saving target class name

2020-10-01 19:38:34 +02:00
parent bafd6be131
commit 62ea95cfe4
4 changed files with 7 additions and 5 deletions
--- a/model_service/views/svm_view.py
+++ b/model_service/views/svm_view.py
@@ -81,7 +81,7 @@ class SVMView(FlaskView):
            os.remove(temp_model_filename)
            os.remove(temp_means_filename)

-        m = AIModel(id=info['id'], type=AIModelType.SVM)
+        m = AIModel(id=info['id'], type=AIModelType.SVM, target_class_name=info['target_class_name'])

        d = SVMDetails(
            aimodel=m,