spacer = (40 - value_len) / 2 print ' ' * spacer, str(value) print '*' * 40 mse = [] models = [] #models.append(ElasticNet(alpha=1, rho=0.7)) models.append(RidgeCV(alphas=[1, 10, 50, 100, 1000])) #models.append(Lasso(alpha=0.1)) #models.append(LinearRegression()) #models.append(SVR(epsilon=3600)) """ for model in models: main(10000, model) """ for x in range(1000, 40100, 100): y_mse, y_training_mse = main(x, models[0]) tuple = (x, y_mse, y_training_mse) mse.append(tuple) general_persistence.dump(mse, 'SVR.out')
import general_persistence as gp def transaction_duration(transaction): """ Based on a list of messages, returns the time elapsed between the first and the last one in seconds. """ start_date = transaction['start_date'] end_date = transaction['end_date'] return (end_date - start_date).total_seconds() db = mdb.connect(host="localhost", user="******", passwd="nolwen", db="watchTowerSpace") db.query("""SELECT priority, start_date, end_date FROM transactions""") rows = db.store_result().fetch_row(0,1) rows = [row for row in rows if row['end_date'] is not None] priorities_dist = {} for row in rows: priority = row['priority'] if priority not in priorities_dist: priorities_dist[priority] = [transaction_duration(row)] else: priorities_dist[priority].append(transaction_duration(row)) gp.dump(priorities_dist, 'out/priority_dist.out')