birbnetes/model-service
4
0
Fork 0
Code Issues Releases Wiki Activity

Added saving target class name
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Pünkösd Marcell
2020-10-01 19:38:34 +02:00
parent bafd6be131
commit 62ea95cfe4
4 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
model_service/views/cnn_view.py
View File

@@ -48,10 +48,7 @@ class CNNView(FlaskView):
ensure_buckets()
# Create the entry in the db
if info['id']:
m = AIModel(id=info['id'], type=AIModelType.CNN)
else:
m = AIModel(type=AIModelType.CNN)
m = AIModel(id=info['id'], type=AIModelType.CNN, target_class_name=info['target_class_name'])
# Put files into MinIO
storage.connection.put_object(current_app.config['MINIO_CNN_BUCKET_NAME'], "model/" + str(m.id), model_file,

2
model_service/views/svm_view.py
View File

@@ -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,