Skip to content

Commit

Permalink
[FIX] Electricity dataset on tutorials (#554)
Browse files Browse the repository at this point in the history
  • Loading branch information
elephaint authored Dec 4, 2024
1 parent 463371e commit c8e0c19
Show file tree
Hide file tree
Showing 5 changed files with 654 additions and 646 deletions.

Large diffs are not rendered by default.

168 changes: 86 additions & 82 deletions nbs/docs/getting-started/4_data_requirements.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -187,8 +187,10 @@
"output_type": "stream",
"text": [
"INFO:nixtla.nixtla_client:Validating inputs...\n",
"INFO:nixtla.nixtla_client:Preprocessing dataframes...\n",
"INFO:nixtla.nixtla_client:Inferred freq: MS\n",
"INFO:nixtla.nixtla_client:Preprocessing dataframes...\n",
"INFO:nixtla.nixtla_client:Querying model metadata...\n",
"INFO:nixtla.nixtla_client:Restricting input...\n",
"INFO:nixtla.nixtla_client:Calling Forecast Endpoint...\n"
]
},
Expand Down Expand Up @@ -221,39 +223,39 @@
" <tr>\n",
" <th>0</th>\n",
" <td>1961-01-01</td>\n",
" <td>437.837921</td>\n",
" <td>437.83792</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>1961-02-01</td>\n",
" <td>426.062714</td>\n",
" <td>426.06270</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>1961-03-01</td>\n",
" <td>463.116547</td>\n",
" <td>463.11655</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>1961-04-01</td>\n",
" <td>478.244507</td>\n",
" <td>478.24450</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>1961-05-01</td>\n",
" <td>505.646484</td>\n",
" <td>505.64648</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" timestamp TimeGPT\n",
"0 1961-01-01 437.837921\n",
"1 1961-02-01 426.062714\n",
"2 1961-03-01 463.116547\n",
"3 1961-04-01 478.244507\n",
"4 1961-05-01 505.646484"
" timestamp TimeGPT\n",
"0 1961-01-01 437.83792\n",
"1 1961-02-01 426.06270\n",
"2 1961-03-01 463.11655\n",
"3 1961-04-01 478.24450\n",
"4 1961-05-01 505.64648"
]
},
"execution_count": null,
Expand Down Expand Up @@ -387,8 +389,10 @@
"output_type": "stream",
"text": [
"INFO:nixtla.nixtla_client:Validating inputs...\n",
"INFO:nixtla.nixtla_client:Inferred freq: h\n",
"INFO:nixtla.nixtla_client:Preprocessing dataframes...\n",
"INFO:nixtla.nixtla_client:Inferred freq: H\n",
"INFO:nixtla.nixtla_client:Querying model metadata...\n",
"INFO:nixtla.nixtla_client:Restricting input...\n",
"INFO:nixtla.nixtla_client:Calling Forecast Endpoint...\n"
]
},
Expand Down Expand Up @@ -423,43 +427,43 @@
" <th>0</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 00:00:00</td>\n",
" <td>45.190453</td>\n",
" <td>45.191200</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 01:00:00</td>\n",
" <td>43.244446</td>\n",
" <td>43.245210</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 02:00:00</td>\n",
" <td>41.958389</td>\n",
" <td>41.959000</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 03:00:00</td>\n",
" <td>39.796486</td>\n",
" <td>39.797043</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 04:00:00</td>\n",
" <td>39.204536</td>\n",
" <td>39.205093</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" unique_id ds TimeGPT\n",
"0 BE 2016-12-31 00:00:00 45.190453\n",
"1 BE 2016-12-31 01:00:00 43.244446\n",
"2 BE 2016-12-31 02:00:00 41.958389\n",
"3 BE 2016-12-31 03:00:00 39.796486\n",
"4 BE 2016-12-31 04:00:00 39.204536"
" unique_id ds TimeGPT\n",
"0 BE 2016-12-31 00:00:00 45.191200\n",
"1 BE 2016-12-31 01:00:00 43.245210\n",
"2 BE 2016-12-31 02:00:00 41.959000\n",
"3 BE 2016-12-31 03:00:00 39.797043\n",
"4 BE 2016-12-31 04:00:00 39.205093"
]
},
"execution_count": null,
Expand Down Expand Up @@ -537,96 +541,96 @@
" <tr>\n",
" <th>0</th>\n",
" <td>BE</td>\n",
" <td>2016-10-22 00:00:00</td>\n",
" <td>70.00</td>\n",
" <td>49593.0</td>\n",
" <td>57253.0</td>\n",
" <td>2011-01-09 00:00:00</td>\n",
" <td>32.54</td>\n",
" <td>63000.0</td>\n",
" <td>63065.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>1.0</td>\n",
" <td>0.0</td>\n",
" <td>1.0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>BE</td>\n",
" <td>2016-10-22 01:00:00</td>\n",
" <td>37.10</td>\n",
" <td>46073.0</td>\n",
" <td>51887.0</td>\n",
" <td>2011-01-09 01:00:00</td>\n",
" <td>21.55</td>\n",
" <td>58800.0</td>\n",
" <td>62715.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>1.0</td>\n",
" <td>0.0</td>\n",
" <td>1.0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>BE</td>\n",
" <td>2016-10-22 02:00:00</td>\n",
" <td>37.10</td>\n",
" <td>44927.0</td>\n",
" <td>51896.0</td>\n",
" <td>2011-01-09 02:00:00</td>\n",
" <td>15.71</td>\n",
" <td>58500.0</td>\n",
" <td>61952.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>1.0</td>\n",
" <td>0.0</td>\n",
" <td>1.0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>BE</td>\n",
" <td>2016-10-22 03:00:00</td>\n",
" <td>44.75</td>\n",
" <td>44483.0</td>\n",
" <td>48428.0</td>\n",
" <td>2011-01-09 03:00:00</td>\n",
" <td>10.58</td>\n",
" <td>54300.0</td>\n",
" <td>59262.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>1.0</td>\n",
" <td>0.0</td>\n",
" <td>1.0</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>BE</td>\n",
" <td>2016-10-22 04:00:00</td>\n",
" <td>37.10</td>\n",
" <td>44338.0</td>\n",
" <td>46721.0</td>\n",
" <td>2011-01-09 04:00:00</td>\n",
" <td>10.32</td>\n",
" <td>51900.0</td>\n",
" <td>56883.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>1.0</td>\n",
" <td>0.0</td>\n",
" <td>1.0</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" unique_id ds y Exogenous1 Exogenous2 day_0 day_1 \\\n",
"0 BE 2016-10-22 00:00:00 70.00 49593.0 57253.0 0.0 0.0 \n",
"1 BE 2016-10-22 01:00:00 37.10 46073.0 51887.0 0.0 0.0 \n",
"2 BE 2016-10-22 02:00:00 37.10 44927.0 51896.0 0.0 0.0 \n",
"3 BE 2016-10-22 03:00:00 44.75 44483.0 48428.0 0.0 0.0 \n",
"4 BE 2016-10-22 04:00:00 37.10 44338.0 46721.0 0.0 0.0 \n",
"0 BE 2011-01-09 00:00:00 32.54 63000.0 63065.0 0.0 0.0 \n",
"1 BE 2011-01-09 01:00:00 21.55 58800.0 62715.0 0.0 0.0 \n",
"2 BE 2011-01-09 02:00:00 15.71 58500.0 61952.0 0.0 0.0 \n",
"3 BE 2011-01-09 03:00:00 10.58 54300.0 59262.0 0.0 0.0 \n",
"4 BE 2011-01-09 04:00:00 10.32 51900.0 56883.0 0.0 0.0 \n",
"\n",
" day_2 day_3 day_4 day_5 day_6 \n",
"0 0.0 0.0 0.0 1.0 0.0 \n",
"1 0.0 0.0 0.0 1.0 0.0 \n",
"2 0.0 0.0 0.0 1.0 0.0 \n",
"3 0.0 0.0 0.0 1.0 0.0 \n",
"4 0.0 0.0 0.0 1.0 0.0 "
"0 0.0 0.0 0.0 0.0 1.0 \n",
"1 0.0 0.0 0.0 0.0 1.0 \n",
"2 0.0 0.0 0.0 0.0 1.0 \n",
"3 0.0 0.0 0.0 0.0 1.0 \n",
"4 0.0 0.0 0.0 0.0 1.0 "
]
},
"execution_count": null,
Expand Down Expand Up @@ -690,8 +694,8 @@
" <th>0</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 00:00:00</td>\n",
" <td>64108.0</td>\n",
" <td>70318.0</td>\n",
" <td>64108.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
Expand All @@ -704,8 +708,8 @@
" <th>1</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 01:00:00</td>\n",
" <td>62492.0</td>\n",
" <td>67898.0</td>\n",
" <td>62492.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
Expand All @@ -718,8 +722,8 @@
" <th>2</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 02:00:00</td>\n",
" <td>61571.0</td>\n",
" <td>68379.0</td>\n",
" <td>61571.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
Expand All @@ -732,8 +736,8 @@
" <th>3</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 03:00:00</td>\n",
" <td>60381.0</td>\n",
" <td>64972.0</td>\n",
" <td>60381.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
Expand All @@ -746,8 +750,8 @@
" <th>4</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 04:00:00</td>\n",
" <td>60298.0</td>\n",
" <td>62900.0</td>\n",
" <td>60298.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
" <td>0.0</td>\n",
Expand All @@ -762,11 +766,11 @@
],
"text/plain": [
" unique_id ds Exogenous1 Exogenous2 day_0 day_1 day_2 \\\n",
"0 BE 2016-12-31 00:00:00 64108.0 70318.0 0.0 0.0 0.0 \n",
"1 BE 2016-12-31 01:00:00 62492.0 67898.0 0.0 0.0 0.0 \n",
"2 BE 2016-12-31 02:00:00 61571.0 68379.0 0.0 0.0 0.0 \n",
"3 BE 2016-12-31 03:00:00 60381.0 64972.0 0.0 0.0 0.0 \n",
"4 BE 2016-12-31 04:00:00 60298.0 62900.0 0.0 0.0 0.0 \n",
"0 BE 2016-12-31 00:00:00 70318.0 64108.0 0.0 0.0 0.0 \n",
"1 BE 2016-12-31 01:00:00 67898.0 62492.0 0.0 0.0 0.0 \n",
"2 BE 2016-12-31 02:00:00 68379.0 61571.0 0.0 0.0 0.0 \n",
"3 BE 2016-12-31 03:00:00 64972.0 60381.0 0.0 0.0 0.0 \n",
"4 BE 2016-12-31 04:00:00 62900.0 60298.0 0.0 0.0 0.0 \n",
"\n",
" day_3 day_4 day_5 day_6 \n",
"0 0.0 0.0 1.0 0.0 \n",
Expand Down Expand Up @@ -796,9 +800,9 @@
"output_type": "stream",
"text": [
"INFO:nixtla.nixtla_client:Validating inputs...\n",
"INFO:nixtla.nixtla_client:Inferred freq: h\n",
"INFO:nixtla.nixtla_client:Preprocessing dataframes...\n",
"INFO:nixtla.nixtla_client:Inferred freq: H\n",
"INFO:nixtla.nixtla_client:Using the following exogenous variables: Exogenous1, Exogenous2, day_0, day_1, day_2, day_3, day_4, day_5, day_6\n",
"INFO:nixtla.nixtla_client:Using future exogenous features: ['Exogenous1', 'Exogenous2', 'day_0', 'day_1', 'day_2', 'day_3', 'day_4', 'day_5', 'day_6']\n",
"INFO:nixtla.nixtla_client:Calling Forecast Endpoint...\n"
]
},
Expand Down Expand Up @@ -833,43 +837,43 @@
" <th>0</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 00:00:00</td>\n",
" <td>74.540773</td>\n",
" <td>46.036015</td>\n",
" </tr>\n",
" <tr>\n",
" <th>1</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 01:00:00</td>\n",
" <td>43.344289</td>\n",
" <td>39.670720</td>\n",
" </tr>\n",
" <tr>\n",
" <th>2</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 02:00:00</td>\n",
" <td>44.429220</td>\n",
" <td>42.782120</td>\n",
" </tr>\n",
" <tr>\n",
" <th>3</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 03:00:00</td>\n",
" <td>38.094395</td>\n",
" <td>33.456580</td>\n",
" </tr>\n",
" <tr>\n",
" <th>4</th>\n",
" <td>BE</td>\n",
" <td>2016-12-31 04:00:00</td>\n",
" <td>37.389141</td>\n",
" <td>33.013958</td>\n",
" </tr>\n",
" </tbody>\n",
"</table>\n",
"</div>"
],
"text/plain": [
" unique_id ds TimeGPT\n",
"0 BE 2016-12-31 00:00:00 74.540773\n",
"1 BE 2016-12-31 01:00:00 43.344289\n",
"2 BE 2016-12-31 02:00:00 44.429220\n",
"3 BE 2016-12-31 03:00:00 38.094395\n",
"4 BE 2016-12-31 04:00:00 37.389141"
" unique_id ds TimeGPT\n",
"0 BE 2016-12-31 00:00:00 46.036015\n",
"1 BE 2016-12-31 01:00:00 39.670720\n",
"2 BE 2016-12-31 02:00:00 42.782120\n",
"3 BE 2016-12-31 03:00:00 33.456580\n",
"4 BE 2016-12-31 04:00:00 33.013958"
]
},
"execution_count": null,
Expand Down
Loading

0 comments on commit c8e0c19

Please sign in to comment.