Fixed Pytest warnings for machine_learning/forecasting (#8958)

* updating DIRECTORY.md

* Fixed pyTest Warnings

---------

Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
This commit is contained in:
Adithya Awati 2023-08-14 14:04:16 +05:30 committed by GitHub
parent 4b7ecb6a81
commit ac68dc1128
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 1 deletions

View File

@ -340,6 +340,7 @@
* [Rod Cutting](dynamic_programming/rod_cutting.py) * [Rod Cutting](dynamic_programming/rod_cutting.py)
* [Subset Generation](dynamic_programming/subset_generation.py) * [Subset Generation](dynamic_programming/subset_generation.py)
* [Sum Of Subset](dynamic_programming/sum_of_subset.py) * [Sum Of Subset](dynamic_programming/sum_of_subset.py)
* [Tribonacci](dynamic_programming/tribonacci.py)
* [Viterbi](dynamic_programming/viterbi.py) * [Viterbi](dynamic_programming/viterbi.py)
* [Word Break](dynamic_programming/word_break.py) * [Word Break](dynamic_programming/word_break.py)

View File

@ -11,6 +11,8 @@ missing (the amount of data that u expected are not supposed to be)
u can just adjust it for ur own purpose u can just adjust it for ur own purpose
""" """
from warnings import simplefilter
import numpy as np import numpy as np
import pandas as pd import pandas as pd
from sklearn.preprocessing import Normalizer from sklearn.preprocessing import Normalizer
@ -45,8 +47,10 @@ def sarimax_predictor(train_user: list, train_match: list, test_match: list) ->
>>> sarimax_predictor([4,2,6,8], [3,1,2,4], [2]) >>> sarimax_predictor([4,2,6,8], [3,1,2,4], [2])
6.6666671111109626 6.6666671111109626
""" """
# Suppress the User Warning raised by SARIMAX due to insufficient observations
simplefilter("ignore", UserWarning)
order = (1, 2, 1) order = (1, 2, 1)
seasonal_order = (1, 1, 0, 7) seasonal_order = (1, 1, 1, 7)
model = SARIMAX( model = SARIMAX(
train_user, exog=train_match, order=order, seasonal_order=seasonal_order train_user, exog=train_match, order=order, seasonal_order=seasonal_order
) )