gavindiaz/drift
1
0
Fork 0
mirror of https://github.com/webclinic017/drift.git synced 2026-08-12 18:38:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(Reporting): only report the last model's results, moved wandb-related functions to reporting (#69)

Browse Source 
* refactor(Reporting): only report the last model's results, moved wandb-related functions to `reporting`

* fix(Reporting): use .mean() on axis 1 to retain the metrics, fixed get_model_name()

* fix(Config): sweep file syntax

* fix(Config): changed hyperparameter search method to "bayes"

* chore(Sweep): adjusted sweep config based on the results we saw (removed Momentum as well)

* fix(Sweep): only use classification method for now, we're not yet prepared for regression
This commit is contained in:
Mark Aron Szulyovszky
2021-12-22 12:04:38 +01:00
committed by GitHub
parent cfc9529274
commit 25b64f5a3d
8 changed files with 78 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/model_map.py
+2 -2
View File
@@ -46,8 +46,8 @@ model_map = {
)
}
model_names_classification = model_map["classification_models"].keys()
model_names_regression = model_map["regression_models"].keys()
model_names_classification = list(model_map["classification_models"].keys())
model_names_regression = list(model_map["regression_models"].keys())
def map_model_name_to_function(model_config:dict, method:str) -> dict: