Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing binding thrombin dataset

......@@ -234,13 +234,17 @@ if __name__ == "__main__":
print(" Done!")
print("Testing (prediction in new data)...")
if args.reduction is not None:
X_test = reduc.transform(X_test)
y_pred = classifier.predict(X_test)
print(" Done!")
print("Saving report...")
with open(os.path.join(args.outputReportPath, args.outputReportFile), mode='w', encoding='utf8') as oFile:
oFile.write('********** EVALUATION REPORT **********\n')
oFile.write('Reduction: {}\n'.format(args.reduction))
oFile.write('Classifier: {}\n'.format(args.classifier))
oFile.write('Kernel: {}\n'.format(args.kernel))
oFile.write('Accuracy: {}\n'.format(accuracy_score(y_test, y_pred)))
oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred, average='weighted')))
oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average='weighted')))
......