X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/predictops.git/blobdiff_plain/b1216db088f3ab2d18dec528332c1e2be3dee78e..1eeee0f8e0d074a2b4453ef5e9f880189d89a64f:/main.py?ds=inline diff --git a/main.py b/main.py index 9ccb687..426d3b5 100644 --- a/main.py +++ b/main.py @@ -1,38 +1,35 @@ -from lib.source import MeteoFrance +from predictops.engine import Engine +from predictops.learn.preprocessing import Preprocessing +from predictops.target.toarea import ToArea 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, clean = False): - logger.info("Predictops engine launched") - if clean: - self.clean() - print("To prevent from downloading again csv files, copy the archive in data rep") + config = (Path.cwd() / 'config') / 'learn.cfg' + engine = Engine(config_file = config) - 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() + engine.add_features() + #print(engine.X) + process = Preprocessing(config_file = config, dict_features = engine.X) - def add_meteofrance(self): - self.meteofrance = MeteoFrance() + #print(engine.X[datetime.strptime('06/30/2010 21:00:00', '%m/%d/%Y %H:%M:%S')]) + print(process.dataframe.head(n=20)) + print(process.dataframe.tail(n=20)) + 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, + csv_file = Path.cwd() / 'data' / 'targets' / 'sdis25' / 'interventions.csv') -engine = Engine(clean = False) -engine.add_meteofrance() -engine.meteofrance.update() -print(len(engine.meteofrance.dated_features)) \ No newline at end of file