Skip to content

Commit

Permalink
feat: 🎸 support empty evaluation horizons
Browse files Browse the repository at this point in the history
  • Loading branch information
zezhishao committed Dec 15, 2023
1 parent fa83069 commit e53374e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion basicts/runners/base_tsf_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ def __init__(self, cfg: Dict):
# evaluation
self.if_evaluate_on_gpu = cfg.get("EVAL", EasyDict()).get("USE_GPU", True) # evaluate on gpu or cpu (gpu is faster but may cause OOM)
self.evaluation_horizons = [_ - 1 for _ in cfg.get("EVAL", EasyDict()).get("HORIZONS", range(1, 13))]
assert min(self.evaluation_horizons) >= 0, "The horizon should start counting from 1."
assert len(self.evaluation_horizons) == 0 or min(self.evaluation_horizons) >= 0, "The horizon should start counting from 1."

def setup_graph(self, cfg: Dict, train: bool):
"""Setup all parameters and the computation graph.
Expand Down

0 comments on commit e53374e

Please sign in to comment.