test_csv['price'] = pd.Series(y_pred, index=test_csv.index)
# any.print_with_line_separator("test_result head():\n" + str(test_csv.head()))
test_csv.to_csv('hw_data_set_2_result.csv', index=False)
df_train = any.read_traning_data()
df_test = any.read_testing_data()
# any.plot_corr(df_train)
# any.plot_base_price(df_train)
df_train = any.clean_data(df_train)
# any.plot_price_log_price(df_train)
# any.plot_zip_hist(df_train)
# any.plot_make_id_hist(df_train)
# any.plot_trim_id_hist(df_train)
any.analyze_data(df_train, df_test)
df_train, df_test, max_seq_engine_index, max_seq_trim_index, max_seq_engine, max_seq_trim = any.date_processing(
df_train, df_test)
max_seq_engine += 5
max_seq_trim += 5
# any.plot_corr(df_train)
# any.plot_eng_seq_hist(df_train)
# any.plot_trim_seq_hist(df_train)
# any.plot_time_hist(df_train)
# any.plot_base_price(df_train)
model, history = train(df_train, df_test, max_seq_engine_index,
max_seq_trim_index, max_seq_engine, max_seq_trim)
# plot_acc(history)
plot_err(history)
predict_test_set(model, df_test, max_seq_engine, max_seq_trim)
