model.py 312 B

12345678910
  1. import numpy as np
  2. import pandas as pd
  3. import statsmodels.api as sm
  4. def smooth(y:pd.Series,frac):
  5. y = y.dropna()
  6. index = y.index
  7. x = np.arange(len(y))
  8. lowess_result = sm.nonparametric.lowess(y, x, frac=frac)
  9. return pd.Series(lowess_result[:, 1], index=index)