-
Notifications
You must be signed in to change notification settings - Fork 0
/
utils.py
21 lines (17 loc) · 736 Bytes
/
utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import numpy as np
import pandas as pd
def save_pred(test_df, y, sub_id):
original_index = test_df.index,
output_df = pd.DataFrame(y, index=original_index, columns=["LABELS"], dtype=int)
output_df.to_csv(f"./submission/submission{sub_id}.csv")
def study_summary(study):
# function to display optuna's best parameters
# not all parameters are displayed, just those from the grid
print("Study: ", study.study_name)
print("\nNumber of finished trials: {}".format(len(study.trials)))
print("Best trial:")
trial = study.best_trial
print(" Validation score: {}".format(trial.value))
print(" Params: ")
for key, val in trial.params.items():
print(" {}: {}".format(key, val))