Carlos-Francisco Méndez-Cruz

Training, crossvalidation and testing structural domain dataset

...@@ -100,7 +100,7 @@ if __name__ == "__main__": ...@@ -100,7 +100,7 @@ if __name__ == "__main__":
100 choices=('linear', 'rbf', 'poly'), default='linear') 100 choices=('linear', 'rbf', 'poly'), default='linear')
101 parser.add_argument("--reduction", dest="reduction", 101 parser.add_argument("--reduction", dest="reduction",
102 help="Feature selection or dimensionality reduction", metavar="NAME", 102 help="Feature selection or dimensionality reduction", metavar="NAME",
103 - choices=('SVD200', 'SVD300', 'CHI250', 'CHI2100'), default='SVD200') 103 + choices=('SVD200', 'SVD300', 'CHI250', 'CHI2100'), default=None)
104 parser.add_argument("--removeStopWords", default=False, 104 parser.add_argument("--removeStopWords", default=False,
105 action="store_true", dest="removeStopWords", 105 action="store_true", dest="removeStopWords",
106 help="Remove stop words") 106 help="Remove stop words")
...@@ -208,13 +208,13 @@ if __name__ == "__main__": ...@@ -208,13 +208,13 @@ 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 A: {}".format(y_train.count('A'))) 211 + print(" Number of training class RI: {}".format(y_train.count('RI')))
212 - print(" Number of training class I: {}".format(y_train.count('I'))) 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 A: {}".format(y_test.count('A'))) 216 + print(" Number of testing class RI: {}".format(y_test.count('RI')))
217 - print(" Number of testing class I: {}".format(y_test.count('I'))) 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
220 # Feature selection and dimensional reduction 220 # Feature selection and dimensional reduction
......