Training, crossvalidation and testing binding thrombin dataset
Showing
1 changed file
with
6 additions
and
3 deletions
... | @@ -246,9 +246,12 @@ if __name__ == "__main__": | ... | @@ -246,9 +246,12 @@ if __name__ == "__main__": |
246 | oFile.write('Classifier: {}\n'.format(args.classifier)) | 246 | oFile.write('Classifier: {}\n'.format(args.classifier)) |
247 | oFile.write('Kernel: {}\n'.format(args.kernel)) | 247 | oFile.write('Kernel: {}\n'.format(args.kernel)) |
248 | oFile.write('Accuracy: {}\n'.format(accuracy_score(y_test, y_pred))) | 248 | oFile.write('Accuracy: {}\n'.format(accuracy_score(y_test, y_pred))) |
249 | - oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred, average='weighted'))) | 249 | + #oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred, average='weighted'))) |
250 | - oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average='weighted'))) | 250 | + #oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average='weighted'))) |
251 | - oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred, average='weighted'))) | 251 | + #oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred, average='weighted'))) |
252 | + oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred))) | ||
253 | + oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred))) | ||
254 | + oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred))) | ||
252 | oFile.write('Confusion matrix: \n') | 255 | oFile.write('Confusion matrix: \n') |
253 | oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n') | 256 | oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n') |
254 | oFile.write('Classification report: \n') | 257 | oFile.write('Classification report: \n') | ... | ... |
-
Please register or login to post a comment