Commit 1977e62f authored by Joaquin Torres's avatar Joaquin Torres

Deleted column in results df

parent 203b75e8
......@@ -143,7 +143,7 @@ if __name__ == "__main__":
# Use group of models with class weight if needed
models = models_CS if j == 1 else models_simple
# Save optimal hyperparameters for each of the models -> metrics will be computed in a different file
hyperparam_df = pd.DataFrame(index=list(models.keys()), columns=['Model Name', 'Best Parameters'])
hyperparam_df = pd.DataFrame(index=list(models.keys()), columns=['Best Parameters'])
for model_name, model in models.items():
print(f"{group}-{method_names[j]}-{model_name}")
# Find optimal hyperparams for curr model
......
Markdown is supported
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