Training, crossvalidation and testing structural domain dataset
Showing
1 changed file
with
5 additions
and
5 deletions
... | @@ -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 | ... | ... |
-
Please register or login to post a comment