1234567891011121314151617181920212223242526272829303132333435363738394041424344 |
- from argparse import ArgumentParser, FileType
- import numpy as np
- import pandas as p
- import statsmodels.formula.api as smf
- import datetime as dt
- import pickle
- from pymodels import thirtyoffset, predweather, harmonic
- from pprint import pprint
-
- def main():
- parser = ArgumentParser(description=
- 'Predict from harmonic model of cluster')
- parser.add_argument("-m", "--model-file", dest="model_file",
- help="filename for models",
- default="../models/testmod.pkl",
- type=FileType('rb'))
- # parser.add_argument("-w", "--weather", dest="weather",
- # help="input weather pickle path",
- # metavar="PATH", required=True,
- # type=FileType('rb'))
- parser.add_argument("--weather-harmonics", dest="weather_harmonics",
- help="number of harmonics for weather; default: 2",
- type=int, default=2, metavar="NUM")
- parser.add_argument("--icp-harmonics", dest="icp_harmonics", nargs=3,
- help="harmonics for icp fitting, default 2 3 3",
- default=[2, 3, 3], type=int,
- metavar="NUM")
- args = parser.parse_args()
-
- print(args)
-
- mods = pickle.load(args.model_file)
-
- pprint(mods)
-
- print(mods["max_temp"].predict())
-
- print(predweather(mods["max_temp"], "2019-01-01", "2019-02-01", harmonics = mods["weather_harmonics"]))
-
- args.model_file.close()
-
-
- if __name__ == "__main__":
- main()
|