Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Hitl dashbaord model viz improve #1333

Open
wants to merge 19 commits into
base: hitl_dashboard_model_mng_fe
Choose a base branch
from
Open
Original file line number Diff line number Diff line change
Expand Up @@ -202,6 +202,7 @@ def get_best_model_loss_acc_by_id(batch_id: int):
- a dict including loss and accuracy for training and valiation, and no error code if can find the best model log
- an error message with error code 404 if cannot find the best model log
"""

def get_best_model_name(batch_id: int):
# download and read best_model.txt file
best_model_fname = ""
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -241,7 +241,7 @@ def get_best_model_loss_acc(batch_id: int):
- a dictionary containing:
- key: loss, value: loss list for training and validation dataset
- key: acc, value: accuracy list for training and validation dataset
- batch id
- batch id
- or an error code indicating the best model log cannot be find
"""
print(
Expand Down
3 changes: 2 additions & 1 deletion droidlet/tools/hitl/utils/read_model_log.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@ def read_model_log_to_list(fname: str):
- Loss changes over epoch increases of training / validation
- Accuracy changes over epoch increases of training / validation
"""

def get_loss_acc(line: str):
words = line.split(" ")
loss, acc = None, None
Expand All @@ -24,7 +25,7 @@ def get_loss_acc(line: str):
acc = None if acc is math.isnan(acc) else acc
return loss, acc

loss_list = []
loss_list = []
acc_list = []
t_key, v_key = "training", "validation"

Expand Down