def test_kfold(): import os from b4msa.classifier import SVC from b4msa.utils import read_data_labels fname = os.path.dirname(__file__) + '/text.json' X, y = read_data_labels(fname, get_klass='klass', get_tweet='text') hy = SVC.predict_kfold(X, y, n_folds=10) for x in hy: assert x in ['POS', 'NEU', 'NEG']
def test_kfold_pool(): import os from b4msa.classifier import SVC from b4msa.utils import read_data_labels from multiprocessing import Pool fname = os.path.dirname(__file__) + '/text.json' X, y = read_data_labels(fname, get_klass='klass', get_tweet='text') pool = Pool(2) hy = SVC.predict_kfold(X, y, n_folds=2, pool=pool) for x in hy: assert x in ['POS', 'NEU', 'NEG'] pool.close()