Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing binding thrombin dataset

...@@ -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')
......