diff --git a/train-eval/Dockerfile b/train-eval/Dockerfile index 5716b93..e8aa1c8 100644 --- a/train-eval/Dockerfile +++ b/train-eval/Dockerfile @@ -5,12 +5,12 @@ RUN apt install python3-pip -y RUN apt install unzip -y RUN apt install git -y -RUN pip install --user numpy -RUN pip install --user pandas -RUN pip install --user torch -RUN pip install --user keras -RUN pip install --user tensorflow -RUN pip install --user scikit-learn +RUN pip install numpy +RUN pip install pandas +RUN pip install torch +RUN pip install keras +RUN pip install tensorflow +RUN pip install scikit-learn # RUN echo "alias kaggle='~/.local/bin/kaggle'" >> ~/.bashrc diff --git a/train-eval/eval.py b/train-eval/eval.py index ff153bd..8368e49 100755 --- a/train-eval/eval.py +++ b/train-eval/eval.py @@ -38,10 +38,16 @@ model.load_state_dict(torch.load('model.pt')) x_test = torch.tensor(X.values, dtype=torch.float32) pred = model(x_test) pred = pred.detach().numpy() -print ("The accuracy is", accuracy_score(Y, np.argmax(pred, axis=1))) -print ("The precission score is ", precision_score(Y, np.argmax(pred, axis=1))) -print ("The recall score is ", recall_score(Y, np.argmax(pred, axis=1))) +acc = accuracy_score(Y, np.argmax(pred, axis=1)) +prec = precision_score(Y, np.argmax(pred, axis=1)) +recall = recall_score(Y, np.argmax(pred, axis=1)) +print ("The accuracy is", acc) +print ("The precission score is ", prec) +print ("The recall score is ", recall) +file = open('metrics.txt', 'w') +file.write(str(acc) + '\t' + str(prec) + '\t' + str(recall)) +file.close() np.savetxt('prediction.tsv', pred, delimiter='\t') \ No newline at end of file