diff --git a/predict.py b/predict.py index cf8b9d0..7dd651c 100644 --- a/predict.py +++ b/predict.py @@ -28,4 +28,6 @@ f1 = f1_score(y_true, y_pred, average='micro') with open('metrics.txt', 'w') as f: f.write(f"Accuracy: {accuracy:.4f}\n") f.write(f"Micro-average Precision: {precision:.4f}\n") - f.write(f"Micro-average F1-score: {f1:.4f}\n") \ No newline at end of file + f.write(f"Micro-average F1-score: {f1:.4f}\n") + +os.system("docker cp metrics.txt . ") \ No newline at end of file