Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing structural domain dataset

...@@ -208,12 +208,12 @@ if __name__ == "__main__": ...@@ -208,12 +208,12 @@ if __name__ == "__main__":
208 print(" Done!") 208 print(" Done!")
209 209
210 print(" Number of training classes: {}".format(len(y_train))) 210 print(" Number of training classes: {}".format(len(y_train)))
211 - print(" Number of training class RI: {}".format(y_train.count('RI'))) 211 + print(" Number of training class RI: {}".format(y_train.count('DOM')))
212 print(" Number of training class OTHER: {}".format(y_train.count('OTHER'))) 212 print(" Number of training class OTHER: {}".format(y_train.count('OTHER')))
213 print(" Shape of training matrix: {}".format(X_train.shape)) 213 print(" Shape of training matrix: {}".format(X_train.shape))
214 214
215 print(" Number of testing classes: {}".format(len(y_test))) 215 print(" Number of testing classes: {}".format(len(y_test)))
216 - print(" Number of testing class RI: {}".format(y_test.count('RI'))) 216 + print(" Number of testing class RI: {}".format(y_test.count('DOM')))
217 print(" Number of testing class OTHER: {}".format(y_test.count('OTHER'))) 217 print(" Number of testing class OTHER: {}".format(y_test.count('OTHER')))
218 print(" Shape of testing matrix: {}".format(X_test.shape)) 218 print(" Shape of testing matrix: {}".format(X_test.shape))
219 219
......