Update README.md

trane-dev · Jul 8, 2023 · 0b58c3a · 0b58c3a
1 parent b8e340a
commit 0b58c3a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -58,6 +58,7 @@ cutoff_strategy = trane.CutoffStrategy(
 )
 time_col = "Date"
 problem_generator = trane.PredictionProblemGenerator(
+ df=data,
  entity_col=entity_col,
  time_col=time_col,
  cutoff_strategy=cutoff_strategy,
@@ -83,4 +84,4 @@ BibTeX entry:
  year={2016},
  organization={IEEE}
 }
-```
+```