Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing dataset

...@@ -246,12 +246,9 @@ if __name__ == "__main__": ...@@ -246,12 +246,9 @@ 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)))
255 oFile.write('Confusion matrix: \n') 252 oFile.write('Confusion matrix: \n')
256 oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n') 253 oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n')
257 oFile.write('Classification report: \n') 254 oFile.write('Classification report: \n')
......
...@@ -292,12 +292,9 @@ if __name__ == "__main__": ...@@ -292,12 +292,9 @@ if __name__ == "__main__":
292 oFile.write('Classifier: {}\n'.format(args.classifier)) 292 oFile.write('Classifier: {}\n'.format(args.classifier))
293 oFile.write('Kernel: {}\n'.format(args.kernel)) 293 oFile.write('Kernel: {}\n'.format(args.kernel))
294 oFile.write('Accuracy: {}\n'.format(accuracy_score(y_test, y_pred))) 294 oFile.write('Accuracy: {}\n'.format(accuracy_score(y_test, y_pred)))
295 - oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred, average=None))) 295 + oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred, average='weighted')))
296 - oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average=None))) 296 + oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred, average='weighted')))
297 - oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred, average=None))) 297 + oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred, average='weighted')))
298 - #oFile.write('Precision: {}\n'.format(precision_score(y_test, y_pred)))
299 - #oFile.write('Recall: {}\n'.format(recall_score(y_test, y_pred)))
300 - #oFile.write('F-score: {}\n'.format(f1_score(y_test, y_pred)))
301 oFile.write('Confusion matrix: \n') 298 oFile.write('Confusion matrix: \n')
302 oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n') 299 oFile.write(str(confusion_matrix(y_test, y_pred)) + '\n')
303 oFile.write('Classification report: \n') 300 oFile.write('Classification report: \n')
......