Commit c0a6c402 authored by Anna Warno's avatar Anna Warno
Browse files

training message corrected

parent eaa4be38
......@@ -89,7 +89,7 @@ def train(target_column, yaml_file="nbeats.yaml"):
torch.save(model.state_dict(), model_path)
msg = {
"metrics": target_column,
"metrics": [target_column],
"forecasting_method": os.environ.get("METHOD", "nbetas"),
"timestamp": int(time.time()) * 1000,
}
......
......@@ -93,7 +93,7 @@ def train(target_column, prediction_length, yaml_file="model.yaml"):
torch.save(tft.state_dict(), model_path)
msg = {
"metrics": target_column,
"metrics": [target_column],
"forecasting_method": os.environ.get("METHOD", "tft"),
"timestamp": int(time.time()) * 1000,
}
......
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