X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/predictops.git/blobdiff_plain/a3cc2abef295b99990b24b1c822319c80a59ffe2..f04d85cc8028a2721d7d5e8ec866ff8022797bb5:/main.py?ds=sidebyside diff --git a/main.py b/main.py index a6a8d74..969054c 100644 --- a/main.py +++ b/main.py @@ -1,71 +1,36 @@ -from lib.source.ephemeris import Ephemeris -from lib.source.meteofrance import MeteoFrance +from predictops.engine import Engine +from predictops.learn.preprocessing import Preprocessing -from datetime import datetime from logging import getLogger from logging.config import fileConfig from pathlib import Path -from shutil import rmtree + +import geopandas as gpd fileConfig((Path.cwd() / 'config') / 'logging.cfg') logger = getLogger() +if __name__ == '__main__': -class Engine: - def __init__(self, start = None, end = None, time_step = None): - logger.info("Predictops engine launched") - self._X = {} - self._Y = {} - - - def clean(self): - # Cleaning the data directory - logger.info("Cleaning and restoring data directory") - directory = Path.cwd() / 'data' - if directory.is_dir(): - rmtree(directory) - p = Path(Path.cwd() / 'data') - p.mkdir() - - - def add_feature(self, name, **kw): - - if name == 'meteofrance': - meteofeature = MeteoFrance(**kw) - meteofeature.update() - dated_features = meteofeature.dated_features - for date in dated_features: - self._X.setdefault(date,{}).update(dated_features[date]) - elif name == 'ephemeris': - ephemerides = Ephemeris(**kw) - dated_features = ephemerides.dated_features - for date in dated_features: - self._X.setdefault(date,{}).update(dated_features[date]) - + config = (Path.cwd() / 'config') / 'learn.cfg' + engine = Engine(config_file = config) - @property - def X(self): - return self._X + engine.add_features() + engine.add_target() - @X.setter - def X(self, x): - self._X = x + process = Preprocessing(config_file = config, dict_features = engine.X) + print(process.dataframe.head(n=20)) + print(process.dataframe.tail(n=20)) -start = datetime.strptime('01/01/2010 00:00:00', '%m/%d/%Y %H:%M:%S') -end = datetime.strptime('12/31/2010 23:00:00', '%m/%d/%Y %H:%M:%S') + #target = toarea(stream_file = Path.cwd() / 'data' / 'targets' / 'sdis25' / 'interventions.csv') -engine = Engine() -engine.add_feature(name = 'meteofrance', - start = start, end = end, - 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']) + 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, + csv_file = Path.cwd() / 'data' / 'targets' / 'sdis25' / 'interventions.csv') -print(engine.X[datetime.strptime('06/30/2010 21:00:00', '%m/%d/%Y %H:%M:%S')]) \ No newline at end of file