Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing dataset

......@@ -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 I: {}".format(y_train.count('I')))
print(" Number of training class A: {}".format(list(y_train).count('A')))
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))
......