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()
