y1_train = [row['BANDWIDTH_TOTAL'] for row in data]
y2_train = [row['MAX_USER'] for row in data]
dates_predict = list(week_range(start_date_predict, end_date_predict, weekday))
model = Model()
model.fit(x_train, y1_train, y2_train)
y1_predict, y2_predict = model.predict(dates_predict)
for date, y1, y2 in zip(dates_predict, y1_predict, y2_predict):
result.add(date, hour, server, y1, y2)
pbar.close()
print()
submission = Submission()
print('Prepare submission...')
with open('data/test_id.csv', 'r') as test_file:
reader = csv.DictReader(test_file)
for row in tqdm(reader):
test_id = row['id']
update_time = datetime.strptime(row['UPDATE_TIME'], '%Y-%m-%d').date()
hour = int(row['HOUR_ID'])
server = row['SERVER_NAME']
bandwidth, max_user = result.get(update_time, hour, server)
submission.add_entry(test_id, bandwidth, max_user)
print()
print('Start write submission...')
submission.write()
