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