Example #1
0
    def _test_basic(self):
        X_train, y_train, X_test, y_test = twitter_reader.toNumpy()

        X_train, y_train = self.get_sub_set_with_size([X_train, y_train], 50)

        clf = LinearSVC(n_estimators=500)
        clf.fit(X_train, y_train)
        y_pred = clf.predict(X_test)
        cm = confusion_matrix(y_test, y_pred)
        print cm
Example #2
0
 def test_ratio(self):
     dataset = twitter_reader.toNumpy()
     for set_size in numpy.arange(100, 1100, 100):
         self.run_ratio(dataset, set_size)
         print