Skip to content

Commit

Permalink
Updating the write_csv_row function to use the item_id instead of ite…
Browse files Browse the repository at this point in the history
…m key as changed in the synthetic dataset classes in PR #286
  • Loading branch information
Danielle Robinson committed Sep 16, 2019
1 parent 4884489 commit ee89af5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/gluonts/dataset/artificial/generate_synthetic.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ def write_csv_row(
if freq_week_start == "W":
freq_week_start = f"W-{week_dict[timestamp.weekday()]}"
timestamp = pd.Timestamp(data["start"], freq=freq_week_start)
item_id = int(data["item"])
item_id = int(data["item_id"])
for j, target in enumerate(data["target"]):
# Using convention that there are no missing values before the start date
if is_missing and j != 0 and j % num_missing == 0:
Expand Down

0 comments on commit ee89af5

Please sign in to comment.