X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/predictops.git/blobdiff_plain/1aa2671e4f047322d6957a58a7b44a568b25d67d..d6469a787c80df2c938f21d4ae107b84213e238f:/main.py diff --git a/main.py b/main.py index b43c188..a42ce0f 100644 --- a/main.py +++ b/main.py @@ -1,8 +1,9 @@ from predictops.source.ephemeris import Ephemeris from predictops.source.meteofrance import MeteoFrance +from predictops.learn.preprocessing import Preprocessing from predictops.target.toarea import ToArea -from datetime import datetime +from datetime import datetime, timedelta from logging import getLogger from logging.config import fileConfig from pathlib import Path @@ -16,7 +17,6 @@ logger = getLogger() class Engine: def __init__(self, start = None, end = None, time_step = None): - logger.info("Predictops engine launched") self._X = {} self._Y = {} @@ -55,7 +55,7 @@ class Engine: self._X = x -start = datetime.strptime('01/01/2010 00:00:00', '%m/%d/%Y %H:%M:%S') +start = datetime.strptime('01/01/2010 01:00:00', '%m/%d/%Y %H:%M:%S') end = datetime.strptime('12/31/2010 23:00:00', '%m/%d/%Y %H:%M:%S') engine = Engine() @@ -64,16 +64,26 @@ engine.add_feature(name = 'meteofrance', latitude = 47.25, longitude = 6.0333, nb_stations = 3, features = ['temperature', 'pressure']) + engine.add_feature(name = 'ephemeris', start = start, end = end, features = ['hour', 'dayInWeek', 'dayInMonth', 'dayInYear', 'weekInYear', 'month', 'year']) -print(engine.X[datetime.strptime('06/30/2010 21:00:00', '%m/%d/%Y %H:%M:%S')]) + +process = Preprocessing(dict_features = engine.X, + start = start, end = end, timestep = timedelta(hours=6)) + + +df = process.dataframe.head(n=20) +#print(engine.X[datetime.strptime('06/30/2010 21:00:00', '%m/%d/%Y %H:%M:%S')]) +print(df) +exit() depts = gpd.read_file( Path.cwd() / 'data' / 'targets' / 'departments' / "departements-20180101.shp") Doubs = depts.loc[depts['nom'] == 'Doubs'].iloc[0] ToArea(area=Doubs.geometry, - start = start, end = end) + start = start, end = end, + csv_file = Path.cwd() / 'data' / 'targets' / 'sdis25' / 'interventions.csv')