Update pages/SVM-Model.py
Browse files- pages/SVM-Model.py +2 -0
pages/SVM-Model.py
CHANGED
|
@@ -82,6 +82,8 @@ if cell_image_file and cytoplasm_mask_file and nuclei_mask_file:
|
|
| 82 |
predicted_class = svm_model.predict(flattened_image)
|
| 83 |
predicted_label = label_encoder.inverse_transform(predicted_class)[0]
|
| 84 |
|
|
|
|
|
|
|
| 85 |
# Make prediction
|
| 86 |
predicted_label = predict_cell_type(svm_model, label_encoder, cell_image, cytoplasm_mask, nuclei_mask)
|
| 87 |
|
|
|
|
| 82 |
predicted_class = svm_model.predict(flattened_image)
|
| 83 |
predicted_label = label_encoder.inverse_transform(predicted_class)[0]
|
| 84 |
|
| 85 |
+
return predicted_label # Return the predicted label
|
| 86 |
+
|
| 87 |
# Make prediction
|
| 88 |
predicted_label = predict_cell_type(svm_model, label_encoder, cell_image, cytoplasm_mask, nuclei_mask)
|
| 89 |
|