Commit 660610bf authored by zpgeng's avatar zpgeng
Browse files


parent 4847e09a
from config import config
import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np
......@@ -70,7 +71,7 @@ def save_logs(hist, output_directory=config['model_dir'], pytorch=False):
row_best_model = hist_df.loc[index_best_model]
df_best_model = pd.DataFrame(data=np.zeros((1, 4), dtype=np.float), index=[0],
columns=['best_model_train_loss', 'best_model_val_loss', 'best_model_train_acc', 'best_model_val_acc')
columns=['best_model_train_loss', 'best_model_val_loss', 'best_model_train_acc', 'best_model_val_acc'])
df_best_model['best_model_train_loss'] = row_best_model['loss']
df_best_model['best_model_val_loss'] = row_best_model['val_loss']
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment