Training, crossvalidation and testing dataset

Carlos-Francisco Méndez-Cruz
Commit a1e1fff92f82f9eb63ae3ec353a3be4368b39571 a1e1fff9 1 parent 78e471e4
Showing 1 changed file with 2 additions and 2 deletions
clasificacion-automatica/binding-thrombin-dataset/imb-training-crossvalidation-testing-binding-thrombin.py
--- a/clasificacion-automatica/binding-thrombin-dataset/imb-training-crossvalidation-testing-binding-thrombin.py
View file @a1e1fff
+++ b/clasificacion-automatica/binding-thrombin-dataset/imb-training-crossvalidation-testing-binding-thrombin.py
View file @a1e1fff
@@ -221,8 +221,8 @@ if __name__ == "__main__":
         print("  After transformtion with {}".format(args.imbalanced))
         print("   Number of training classes: {}".format(len(y_train)))
-        print("   Number of training class A: {}".format(y_train.count('A')))
+        print("   Number of training class A: {}".format(list(y_train).count('A')))
-        print("   Number of training class I: {}".format(y_train.count('I')))
+        print("   Number of training class I: {}".format(list(y_train).count('I')))
         print("   Shape of training matrix: {}".format(X_train.shape))
         print("      Data transformation done in : %fs" % (time() - t1))