def with_games(name, query, path, method='skip_1'):
'''Trains a model games that have already been scraped.'''
begin = time.time()
dts.build_and_dump(name, query, method=method, path=path)
#dumps model
trn.train_and_save(name, 'logistic_regression')
tst.irun(name, query, 3)
print 'Time:', time.time()-begin
def simple_test(name, query, method='skip_1'):
'''Builds, train, and interactively test a simple model with few games,
for debugging.'''
urls = scr.getURLs('20151128')
dts.build_and_dump(name, query, method=method, urls=urls)
# dumps model
trn.train_and_save(name, 'logistic_regression')
tst.irun(name, query, 3)
def full_test(name, query, train_set, test_set, method):
'''Performs a full automated test.
@param name: test and model name, arbitrary
@param query: question to train on
@param train_set: name of the training set of games
@param test_set: name of testing set of games
@param method: feature extraction method
'''
path = 'games/{}'.format(train_set)
begin = time.time()
dts.build_and_dump(name, query, method=method, path=path)
trn.train_and_save(name, 'logistic_regression')
print time.time()-begin
return tst.test(name, test_set, query, method)
