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