diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index 86bed8eb..3afa27c5 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -79,7 +79,7 @@ jobs: python-version: ${{ matrix.python-version }} - name: Install pip requirements - run: pip install uv && uv pip install --system ".[dev]" && pip install matplotlib + run: pip install uv && uv pip install --system ".[dev]" - name: Install OpenMP if: startsWith(matrix.os, 'macos') diff --git a/nbs/assets/arima_rst.csv b/nbs/assets/arima_rst.csv new file mode 100644 index 00000000..564a3405 --- /dev/null +++ b/nbs/assets/arima_rst.csv @@ -0,0 +1,197 @@ +,unique_id,ds,AutoARIMA +0,FOODS_1,2016-04-25,2567.6380220997385 +1,FOODS_1,2016-04-26,2640.0126507600617 +2,FOODS_1,2016-04-27,2654.020161270193 +3,FOODS_1,2016-04-28,2796.440947432241 +4,FOODS_1,2016-04-29,3208.3811781219893 +5,FOODS_1,2016-04-30,3773.861663861278 +6,FOODS_1,2016-05-01,3299.79743620078 +7,FOODS_1,2016-05-02,2640.917833365884 +8,FOODS_1,2016-05-03,2694.336636469822 +9,FOODS_1,2016-05-04,2694.1035608286916 +10,FOODS_1,2016-05-05,2835.3163066249176 +11,FOODS_1,2016-05-06,3236.293019356992 +12,FOODS_1,2016-05-07,3797.241666185409 +13,FOODS_1,2016-05-08,3321.3663551719346 +14,FOODS_1,2016-05-09,2657.5206296593224 +15,FOODS_1,2016-05-10,2707.524732902354 +16,FOODS_1,2016-05-11,2704.747898679939 +17,FOODS_1,2016-05-12,2843.9075334840127 +18,FOODS_1,2016-05-13,3243.2271452721875 +19,FOODS_1,2016-05-14,3802.8383176295374 +20,FOODS_1,2016-05-15,3325.8835081296597 +21,FOODS_1,2016-05-16,2661.1665009534677 +22,FOODS_1,2016-05-17,2710.4673778864567 +23,FOODS_1,2016-05-18,2707.1229576998144 +24,FOODS_1,2016-05-19,2845.8244841791743 +25,FOODS_1,2016-05-20,3244.7743489150994 +26,FOODS_1,2016-05-21,3804.0870921121996 +27,FOODS_1,2016-05-22,3326.8914153406417 +28,FOODS_2,2016-04-25,5247.139313835306 +29,FOODS_2,2016-04-26,4792.960953564678 +30,FOODS_2,2016-04-27,4590.964924105041 +31,FOODS_2,2016-04-28,4600.860960731239 +32,FOODS_2,2016-04-29,4942.967875785429 +33,FOODS_2,2016-04-30,6337.5344008647035 +34,FOODS_2,2016-05-01,6757.903656868389 +35,FOODS_2,2016-05-02,5607.448278542931 +36,FOODS_2,2016-05-03,5244.736379588703 +37,FOODS_2,2016-05-04,5004.324139775963 +38,FOODS_2,2016-05-05,4898.957821332703 +39,FOODS_2,2016-05-06,5310.50858073175 +40,FOODS_2,2016-05-07,6641.501751486881 +41,FOODS_2,2016-05-08,6994.60266694966 +42,FOODS_2,2016-05-09,5779.319357423172 +43,FOODS_2,2016-05-10,5407.162676474501 +44,FOODS_2,2016-05-11,5083.913842178118 +45,FOODS_2,2016-05-12,4944.398003219373 +46,FOODS_2,2016-05-13,5311.376375871703 +47,FOODS_2,2016-05-14,6593.071399563353 +48,FOODS_2,2016-05-15,6919.214380455565 +49,FOODS_2,2016-05-16,5687.59113569716 +50,FOODS_2,2016-05-17,5289.695188706264 +51,FOODS_2,2016-05-18,4961.060422560739 +52,FOODS_2,2016-05-19,4815.1162421391555 +53,FOODS_2,2016-05-20,5184.318070407874 +54,FOODS_2,2016-05-21,6467.664979048176 +55,FOODS_2,2016-05-22,6807.52592255971 +56,FOODS_3,2016-04-25,17666.731556718732 +57,FOODS_3,2016-04-26,16540.541376421595 +58,FOODS_3,2016-04-27,16160.633071055223 +59,FOODS_3,2016-04-28,16370.683658640599 +60,FOODS_3,2016-04-29,17889.869101221244 +61,FOODS_3,2016-04-30,22332.436097780206 +62,FOODS_3,2016-05-01,22713.044590532798 +63,FOODS_3,2016-05-02,17468.426315986457 +64,FOODS_3,2016-05-03,16293.254187261948 +65,FOODS_3,2016-05-04,15975.967829497 +66,FOODS_3,2016-05-05,16301.448695311468 +67,FOODS_3,2016-05-06,17742.192012594733 +68,FOODS_3,2016-05-07,22149.920466723906 +69,FOODS_3,2016-05-08,22672.51910973226 +70,FOODS_3,2016-05-09,17388.755964833897 +71,FOODS_3,2016-05-10,16229.554742510712 +72,FOODS_3,2016-05-11,15913.853788834986 +73,FOODS_3,2016-05-12,16229.441847880662 +74,FOODS_3,2016-05-13,17695.119173018684 +75,FOODS_3,2016-05-14,22118.38914081269 +76,FOODS_3,2016-05-15,22622.383902927766 +77,FOODS_3,2016-05-16,17352.915739873653 +78,FOODS_3,2016-05-17,16196.863301084244 +79,FOODS_3,2016-05-18,15886.258449760904 +80,FOODS_3,2016-05-19,16208.436163712715 +81,FOODS_3,2016-05-20,17673.61997937043 +82,FOODS_3,2016-05-21,22097.648256558292 +83,FOODS_3,2016-05-22,22608.340998500338 +84,HOBBIES_1,2016-04-25,3301.1184614493914 +85,HOBBIES_1,2016-04-26,3190.0305707575 +86,HOBBIES_1,2016-04-27,3199.935172884026 +87,HOBBIES_1,2016-04-28,3225.695752897986 +88,HOBBIES_1,2016-04-29,3567.265955094044 +89,HOBBIES_1,2016-04-30,4356.224282294094 +90,HOBBIES_1,2016-05-01,4089.925888720474 +91,HOBBIES_1,2016-05-02,3256.234765934616 +92,HOBBIES_1,2016-05-03,3167.254279226249 +93,HOBBIES_1,2016-05-04,3182.164242104323 +94,HOBBIES_1,2016-05-05,3197.8208347975133 +95,HOBBIES_1,2016-05-06,3562.1759714192 +96,HOBBIES_1,2016-05-07,4348.942474784928 +97,HOBBIES_1,2016-05-08,4062.088166835995 +98,HOBBIES_1,2016-05-09,3270.411197190324 +99,HOBBIES_1,2016-05-10,3153.9826259751076 +100,HOBBIES_1,2016-05-11,3185.763508441231 +101,HOBBIES_1,2016-05-12,3215.5124943559786 +102,HOBBIES_1,2016-05-13,3557.0053751276196 +103,HOBBIES_1,2016-05-14,4347.883194155448 +104,HOBBIES_1,2016-05-15,4072.703812696276 +105,HOBBIES_1,2016-05-16,3257.5290353634405 +106,HOBBIES_1,2016-05-17,3160.8541642155105 +107,HOBBIES_1,2016-05-18,3181.513304571057 +108,HOBBIES_1,2016-05-19,3202.9504121022114 +109,HOBBIES_1,2016-05-20,3559.2179367013387 +110,HOBBIES_1,2016-05-21,4348.099858357742 +111,HOBBIES_1,2016-05-22,4064.243078853568 +112,HOBBIES_2,2016-04-25,372.73314373276685 +113,HOBBIES_2,2016-04-26,394.9539049862272 +114,HOBBIES_2,2016-04-27,383.16517087090995 +115,HOBBIES_2,2016-04-28,383.62499302365416 +116,HOBBIES_2,2016-04-29,398.311355639119 +117,HOBBIES_2,2016-04-30,394.347605737723 +118,HOBBIES_2,2016-05-01,410.1164983512131 +119,HOBBIES_2,2016-05-02,381.5187849900045 +120,HOBBIES_2,2016-05-03,390.5415783971641 +121,HOBBIES_2,2016-05-04,384.30591664608176 +122,HOBBIES_2,2016-05-05,382.0656727697983 +123,HOBBIES_2,2016-05-06,391.0787416473706 +124,HOBBIES_2,2016-05-07,388.7836691251789 +125,HOBBIES_2,2016-05-08,393.876028907937 +126,HOBBIES_2,2016-05-09,390.2092112439395 +127,HOBBIES_2,2016-05-10,389.2354081364607 +128,HOBBIES_2,2016-05-11,388.6840382471152 +129,HOBBIES_2,2016-05-12,388.2380749588003 +130,HOBBIES_2,2016-05-13,387.8486951350242 +131,HOBBIES_2,2016-05-14,387.50441758710645 +132,HOBBIES_2,2016-05-15,387.19942014913687 +133,HOBBIES_2,2016-05-16,386.9291391176742 +134,HOBBIES_2,2016-05-17,386.6896116476871 +135,HOBBIES_2,2016-05-18,386.4773369473272 +136,HOBBIES_2,2016-05-19,386.28921406142393 +137,HOBBIES_2,2016-05-20,386.12249505879066 +138,HOBBIES_2,2016-05-21,385.9747446856811 +139,HOBBIES_2,2016-05-22,385.84380476581924 +140,HOUSEHOLD_1,2016-04-25,7401.891643264893 +141,HOUSEHOLD_1,2016-04-26,6713.059783961793 +142,HOUSEHOLD_1,2016-04-27,6568.574826414228 +143,HOUSEHOLD_1,2016-04-28,6780.766120981868 +144,HOUSEHOLD_1,2016-04-29,7572.471090810987 +145,HOUSEHOLD_1,2016-04-30,9760.255572651784 +146,HOUSEHOLD_1,2016-05-01,9626.516172187357 +147,HOUSEHOLD_1,2016-05-02,7339.549091656914 +148,HOUSEHOLD_1,2016-05-03,6715.501042329869 +149,HOUSEHOLD_1,2016-05-04,6588.688826450386 +150,HOUSEHOLD_1,2016-05-05,6792.018975120889 +151,HOUSEHOLD_1,2016-05-06,7585.826196179538 +152,HOUSEHOLD_1,2016-05-07,9784.576440586516 +153,HOUSEHOLD_1,2016-05-08,9625.229656671552 +154,HOUSEHOLD_1,2016-05-09,7370.017398852801 +155,HOUSEHOLD_1,2016-05-10,6703.274498285182 +156,HOUSEHOLD_1,2016-05-11,6599.658566441017 +157,HOUSEHOLD_1,2016-05-12,6803.892952779315 +158,HOUSEHOLD_1,2016-05-13,7578.147413631544 +159,HOUSEHOLD_1,2016-05-14,9797.63840894531 +160,HOUSEHOLD_1,2016-05-15,9618.004359773748 +161,HOUSEHOLD_1,2016-05-16,7369.325267410535 +162,HOUSEHOLD_1,2016-05-17,6704.177617699504 +163,HOUSEHOLD_1,2016-05-18,6602.321156728928 +164,HOUSEHOLD_1,2016-05-19,6806.125635799577 +165,HOUSEHOLD_1,2016-05-20,7579.715540310798 +166,HOUSEHOLD_1,2016-05-21,9800.535564906138 +167,HOUSEHOLD_1,2016-05-22,9618.816928719863 +168,HOUSEHOLD_2,2016-04-25,1944.6365790975688 +169,HOUSEHOLD_2,2016-04-26,1782.1136284351692 +170,HOUSEHOLD_2,2016-04-27,1783.3160335243492 +171,HOUSEHOLD_2,2016-04-28,1843.7479525740052 +172,HOUSEHOLD_2,2016-04-29,2017.7306193648328 +173,HOUSEHOLD_2,2016-04-30,2573.4319970333827 +174,HOUSEHOLD_2,2016-05-01,2555.781560345023 +175,HOUSEHOLD_2,2016-05-02,1911.5887343314832 +176,HOUSEHOLD_2,2016-05-03,1773.6009268551322 +177,HOUSEHOLD_2,2016-05-04,1783.737933427243 +178,HOUSEHOLD_2,2016-05-05,1836.4433591054317 +179,HOUSEHOLD_2,2016-05-06,2026.2345730343163 +180,HOUSEHOLD_2,2016-05-07,2599.273470339867 +181,HOUSEHOLD_2,2016-05-08,2547.324959977864 +182,HOUSEHOLD_2,2016-05-09,1919.6860442557204 +183,HOUSEHOLD_2,2016-05-10,1772.1267493740079 +184,HOUSEHOLD_2,2016-05-11,1776.2903338158771 +185,HOUSEHOLD_2,2016-05-12,1836.0676051839982 +186,HOUSEHOLD_2,2016-05-13,2012.270429155826 +187,HOUSEHOLD_2,2016-05-14,2570.3606621455087 +188,HOUSEHOLD_2,2016-05-15,2548.3296441260773 +189,HOUSEHOLD_2,2016-05-16,1906.4143149755678 +190,HOUSEHOLD_2,2016-05-17,1767.280129446266 +191,HOUSEHOLD_2,2016-05-18,1776.7401745381599 +192,HOUSEHOLD_2,2016-05-19,1830.4754939798 +193,HOUSEHOLD_2,2016-05-20,2018.588737506113 +194,HOUSEHOLD_2,2016-05-21,2589.769743131061 +195,HOUSEHOLD_2,2016-05-22,2541.838195247957 diff --git a/nbs/assets/nhits_rst.csv b/nbs/assets/nhits_rst.csv new file mode 100644 index 00000000..99e3f9d1 --- /dev/null +++ b/nbs/assets/nhits_rst.csv @@ -0,0 +1,197 @@ +unique_id,ds,NHITS +FOODS_1,2016-04-25,2504.762 +FOODS_1,2016-04-26,2467.719 +FOODS_1,2016-04-27,2430.538 +FOODS_1,2016-04-28,2545.7861 +FOODS_1,2016-04-29,3158.0547 +FOODS_1,2016-04-30,3675.847 +FOODS_1,2016-05-01,3220.2334 +FOODS_1,2016-05-02,2603.3481 +FOODS_1,2016-05-03,2549.2483 +FOODS_1,2016-05-04,2526.7446 +FOODS_1,2016-05-05,2656.253 +FOODS_1,2016-05-06,3305.013 +FOODS_1,2016-05-07,3765.003 +FOODS_1,2016-05-08,3275.3057 +FOODS_1,2016-05-09,2697.433 +FOODS_1,2016-05-10,2621.8018 +FOODS_1,2016-05-11,2528.0818 +FOODS_1,2016-05-12,2711.9429 +FOODS_1,2016-05-13,3265.7722 +FOODS_1,2016-05-14,3737.7068 +FOODS_1,2016-05-15,3250.004 +FOODS_1,2016-05-16,2663.8232 +FOODS_1,2016-05-17,2565.505 +FOODS_1,2016-05-18,2516.273 +FOODS_1,2016-05-19,2611.7502 +FOODS_1,2016-05-20,3218.6646 +FOODS_1,2016-05-21,3715.1863 +FOODS_1,2016-05-22,3209.3425 +FOODS_2,2016-04-25,5217.8647 +FOODS_2,2016-04-26,4537.3613 +FOODS_2,2016-04-27,4325.717 +FOODS_2,2016-04-28,4522.6455 +FOODS_2,2016-04-29,5086.342 +FOODS_2,2016-04-30,6492.3867 +FOODS_2,2016-05-01,7200.9097 +FOODS_2,2016-05-02,6160.0244 +FOODS_2,2016-05-03,5388.8823 +FOODS_2,2016-05-04,5440.1694 +FOODS_2,2016-05-05,5520.1226 +FOODS_2,2016-05-06,6055.701 +FOODS_2,2016-05-07,7401.1245 +FOODS_2,2016-05-08,7905.652 +FOODS_2,2016-05-09,6759.285 +FOODS_2,2016-05-10,5789.94 +FOODS_2,2016-05-11,5636.6777 +FOODS_2,2016-05-12,5507.3813 +FOODS_2,2016-05-13,5849.2007 +FOODS_2,2016-05-14,7064.666 +FOODS_2,2016-05-15,7434.761 +FOODS_2,2016-05-16,6070.2705 +FOODS_2,2016-05-17,5095.434 +FOODS_2,2016-05-18,4870.112 +FOODS_2,2016-05-19,4596.719 +FOODS_2,2016-05-20,4982.272 +FOODS_2,2016-05-21,6231.084 +FOODS_2,2016-05-22,6697.241 +FOODS_3,2016-04-25,17230.807 +FOODS_3,2016-04-26,15789.233 +FOODS_3,2016-04-27,15369.597 +FOODS_3,2016-04-28,15617.65 +FOODS_3,2016-04-29,18008.902 +FOODS_3,2016-04-30,23040.523 +FOODS_3,2016-05-01,23571.592 +FOODS_3,2016-05-02,18340.871 +FOODS_3,2016-05-03,16562.752 +FOODS_3,2016-05-04,16747.717 +FOODS_3,2016-05-05,17052.62 +FOODS_3,2016-05-06,19778.832 +FOODS_3,2016-05-07,24705.719 +FOODS_3,2016-05-08,24711.145 +FOODS_3,2016-05-09,19328.555 +FOODS_3,2016-05-10,17409.87 +FOODS_3,2016-05-11,17146.104 +FOODS_3,2016-05-12,16995.588 +FOODS_3,2016-05-13,19478.918 +FOODS_3,2016-05-14,24267.672 +FOODS_3,2016-05-15,24103.676 +FOODS_3,2016-05-16,18767.125 +FOODS_3,2016-05-17,16687.102 +FOODS_3,2016-05-18,16438.53 +FOODS_3,2016-05-19,15937.973 +FOODS_3,2016-05-20,18391.8 +FOODS_3,2016-05-21,23098.781 +FOODS_3,2016-05-22,23017.9 +HOBBIES_1,2016-04-25,3293.069 +HOBBIES_1,2016-04-26,3175.8167 +HOBBIES_1,2016-04-27,3104.7383 +HOBBIES_1,2016-04-28,3130.194 +HOBBIES_1,2016-04-29,3652.3372 +HOBBIES_1,2016-04-30,4440.7046 +HOBBIES_1,2016-05-01,4260.2666 +HOBBIES_1,2016-05-02,3375.5198 +HOBBIES_1,2016-05-03,3186.3364 +HOBBIES_1,2016-05-04,3148.969 +HOBBIES_1,2016-05-05,3190.6125 +HOBBIES_1,2016-05-06,3787.1 +HOBBIES_1,2016-05-07,4503.3115 +HOBBIES_1,2016-05-08,4216.017 +HOBBIES_1,2016-05-09,3309.7373 +HOBBIES_1,2016-05-10,3130.4082 +HOBBIES_1,2016-05-11,3052.7866 +HOBBIES_1,2016-05-12,3083.465 +HOBBIES_1,2016-05-13,3624.693 +HOBBIES_1,2016-05-14,4420.6533 +HOBBIES_1,2016-05-15,4138.2466 +HOBBIES_1,2016-05-16,3327.1152 +HOBBIES_1,2016-05-17,3115.9226 +HOBBIES_1,2016-05-18,3068.1096 +HOBBIES_1,2016-05-19,3066.814 +HOBBIES_1,2016-05-20,3670.4724 +HOBBIES_1,2016-05-21,4458.9883 +HOBBIES_1,2016-05-22,4195.486 +HOBBIES_2,2016-04-25,359.69662 +HOBBIES_2,2016-04-26,337.56714 +HOBBIES_2,2016-04-27,336.36066 +HOBBIES_2,2016-04-28,346.82523 +HOBBIES_2,2016-04-29,385.13193 +HOBBIES_2,2016-04-30,448.0414 +HOBBIES_2,2016-05-01,428.58112 +HOBBIES_2,2016-05-02,347.9329 +HOBBIES_2,2016-05-03,325.00256 +HOBBIES_2,2016-05-04,332.32742 +HOBBIES_2,2016-05-05,348.48566 +HOBBIES_2,2016-05-06,397.46094 +HOBBIES_2,2016-05-07,463.35196 +HOBBIES_2,2016-05-08,445.56622 +HOBBIES_2,2016-05-09,360.51288 +HOBBIES_2,2016-05-10,336.52014 +HOBBIES_2,2016-05-11,344.28247 +HOBBIES_2,2016-05-12,358.7079 +HOBBIES_2,2016-05-13,409.6697 +HOBBIES_2,2016-05-14,475.46484 +HOBBIES_2,2016-05-15,455.52716 +HOBBIES_2,2016-05-16,381.3134 +HOBBIES_2,2016-05-17,355.26096 +HOBBIES_2,2016-05-18,351.80453 +HOBBIES_2,2016-05-19,363.197 +HOBBIES_2,2016-05-20,404.1736 +HOBBIES_2,2016-05-21,471.72308 +HOBBIES_2,2016-05-22,452.07477 +HOUSEHOLD_1,2016-04-25,7322.5625 +HOUSEHOLD_1,2016-04-26,6919.1865 +HOUSEHOLD_1,2016-04-27,6807.973 +HOUSEHOLD_1,2016-04-28,6930.8086 +HOUSEHOLD_1,2016-04-29,8116.5 +HOUSEHOLD_1,2016-04-30,10300.233 +HOUSEHOLD_1,2016-05-01,10267.285 +HOUSEHOLD_1,2016-05-02,7921.134 +HOUSEHOLD_1,2016-05-03,7284.6235 +HOUSEHOLD_1,2016-05-04,7280.303 +HOUSEHOLD_1,2016-05-05,7267.668 +HOUSEHOLD_1,2016-05-06,8540.003 +HOUSEHOLD_1,2016-05-07,10456.705 +HOUSEHOLD_1,2016-05-08,10088.888 +HOUSEHOLD_1,2016-05-09,7604.306 +HOUSEHOLD_1,2016-05-10,6985.8867 +HOUSEHOLD_1,2016-05-11,6745.779 +HOUSEHOLD_1,2016-05-12,6693.4326 +HOUSEHOLD_1,2016-05-13,7825.8516 +HOUSEHOLD_1,2016-05-14,9895.505 +HOUSEHOLD_1,2016-05-15,9434.72 +HOUSEHOLD_1,2016-05-16,7175.374 +HOUSEHOLD_1,2016-05-17,6486.9434 +HOUSEHOLD_1,2016-05-18,6495.8496 +HOUSEHOLD_1,2016-05-19,6345.693 +HOUSEHOLD_1,2016-05-20,7672.4624 +HOUSEHOLD_1,2016-05-21,9825.944 +HOUSEHOLD_1,2016-05-22,9556.082 +HOUSEHOLD_2,2016-04-25,1931.113 +HOUSEHOLD_2,2016-04-26,1782.4025 +HOUSEHOLD_2,2016-04-27,1758.2009 +HOUSEHOLD_2,2016-04-28,1763.3115 +HOUSEHOLD_2,2016-04-29,2031.7482 +HOUSEHOLD_2,2016-04-30,2627.6375 +HOUSEHOLD_2,2016-05-01,2618.0635 +HOUSEHOLD_2,2016-05-02,1939.3312 +HOUSEHOLD_2,2016-05-03,1775.9213 +HOUSEHOLD_2,2016-05-04,1784.6147 +HOUSEHOLD_2,2016-05-05,1768.6473 +HOUSEHOLD_2,2016-05-06,2086.8005 +HOUSEHOLD_2,2016-05-07,2645.3496 +HOUSEHOLD_2,2016-05-08,2584.8975 +HOUSEHOLD_2,2016-05-09,1910.047 +HOUSEHOLD_2,2016-05-10,1752.4678 +HOUSEHOLD_2,2016-05-11,1736.5 +HOUSEHOLD_2,2016-05-12,1733.4746 +HOUSEHOLD_2,2016-05-13,2037.8102 +HOUSEHOLD_2,2016-05-14,2586.608 +HOUSEHOLD_2,2016-05-15,2516.9414 +HOUSEHOLD_2,2016-05-16,1902.0074 +HOUSEHOLD_2,2016-05-17,1732.1702 +HOUSEHOLD_2,2016-05-18,1742.648 +HOUSEHOLD_2,2016-05-19,1724.6858 +HOUSEHOLD_2,2016-05-20,2035.2493 +HOUSEHOLD_2,2016-05-21,2608.6196 +HOUSEHOLD_2,2016-05-22,2549.3857 diff --git a/nbs/docs/getting-started/7_why_timegpt.ipynb b/nbs/docs/getting-started/7_why_timegpt.ipynb index d4f23db0..d2f2bce3 100644 --- a/nbs/docs/getting-started/7_why_timegpt.ipynb +++ b/nbs/docs/getting-started/7_why_timegpt.ipynb @@ -393,8 +393,8 @@ "data": { "text/plain": [ "metric\n", - "rmse 592.599619\n", - "smape 0.049404\n", + "rmse 592.588597\n", + "smape 0.049403\n", "Name: TimeGPT, dtype: float64" ] }, @@ -438,6 +438,7 @@ "metadata": {}, "outputs": [], "source": [ + "#| eval: false\n", "from statsforecast import StatsForecast\n", "from statsforecast.models import AutoARIMA" ] @@ -446,18 +447,10 @@ "cell_type": "code", "execution_count": null, "metadata": {}, - "outputs": [ - { - "name": "stderr", - "output_type": "stream", - "text": [ - "/Users/yibeihu/opt/anaconda3/envs/report/lib/python3.9/site-packages/statsforecast/core.py:494: FutureWarning: In a future version the predictions will have the id as a column. You can set the `NIXTLA_ID_AS_COL` environment variable to adopt the new behavior and to suppress this warning.\n", - " warnings.warn(\n" - ] - } - ], + "outputs": [], "source": [ - "#Initiate ARIMA model\n", + "#| eval: false\n", + "Initiate ARIMA model\n", "sf = StatsForecast(\n", " models=[AutoARIMA(season_length=7)],\n", " freq='D'\n", @@ -466,6 +459,16 @@ "fcst_arima = sf.forecast(h=28, df=df_train) " ] }, + { + "cell_type": "code", + "execution_count": null, + "metadata": {}, + "outputs": [], + "source": [ + "#| hide\n", + "fcst_arima = pd.read_csv('../../assets/arima_rst.csv', parse_dates=['ds'])" + ] + }, { "cell_type": "code", "execution_count": null, @@ -605,6 +608,7 @@ "metadata": {}, "outputs": [], "source": [ + "#| eval: false\n", "from neuralforecast.core import NeuralForecast\n", "from neuralforecast.models import NHITS" ] @@ -613,108 +617,10 @@ "cell_type": "code", "execution_count": null, "metadata": {}, - "outputs": [ - { - "name": "stderr", - "output_type": "stream", - "text": [ - "INFO:lightning_fabric.utilities.seed:Seed set to 1\n", - "INFO:pytorch_lightning.utilities.rank_zero:GPU available: True (mps), used: True\n", - "INFO:pytorch_lightning.utilities.rank_zero:TPU available: False, using: 0 TPU cores\n", - "INFO:pytorch_lightning.utilities.rank_zero:HPU available: False, using: 0 HPUs\n", - "INFO:pytorch_lightning.callbacks.model_summary:\n", - " | Name | Type | Params | Mode \n", - "-------------------------------------------------------\n", - "0 | loss | MAE | 0 | train\n", - "1 | padder_train | ConstantPad1d | 0 | train\n", - "2 | scaler | TemporalNorm | 0 | train\n", - "3 | blocks | ModuleList | 2.5 M | train\n", - "-------------------------------------------------------\n", - "2.5 M Trainable params\n", - "0 Non-trainable params\n", - "2.5 M Total params\n", - "9.849 Total estimated model params size (MB)\n", - "34 Modules in train mode\n", - "0 Modules in eval mode\n" - ] - }, - { - "data": { - "application/vnd.jupyter.widget-view+json": { - "model_id": "a9f134529af4422bb8849fc690e5d486", - "version_major": 2, - "version_minor": 0 - }, - "text/plain": [ - "Sanity Checking: | | 0/? [00:00