def setUp(self):
self.data_loader = AllStateDataLoader()
model = grid_search.GridSearchCV(log, parameters, verbose=verbose)
model.fit(X, y)
joblib.dump(model, filename)
return model
# fitting models
parameters = {
'C': [0.1, 0.5, 1.0],
'loss': ['l2'],
'penalty': ['l1', 'l2'],
'dual': [False]
}
l = AllStateDataLoader()
def get_model_filename(type_dataset, objective_letter, real_letters):
if real_letters == "":
return os.path.join(
"model_linearsvc",
"model_linearsvc_data_%s_%s_without_real_cascade.pkl" %
(type_dataset, objective_letter))
else:
return os.path.join(
"model_linearsvc",
"model_linearsvc_data_%s_%s_with_real_%s_cascade.pkl" %
(type_dataset, objective_letter, real_letters))
def __init__(self):
self.__datasets = {}
self.__dataloader = AllStateDataLoader()
self.debug = True
