Improving csv -> dataframe module

[predictops.git] / main.py

diff --git a/main.py b/main.py
index fe8ed203fe9fd78a2f61b98fed7a0474a4b3d3bd..a42ce0f11cb9b42feae17fa17e1abcdf513b8570 100644 (file)
--- a/main.py
+++ b/main.py
@@ -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,18 +64,20 @@ 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'])
 
+
 process = Preprocessing(dict_features = engine.X,
-                   start = start, end = end, timestep = timedelta(hours=1))
+                   start = start, end = end, timestep = timedelta(hours=6))
 
-process.fill_na()
-print(process.dataframe.head(n=20))
-#print(engine.X[datetime.strptime('06/30/2010 21:00:00', '%m/%d/%Y %H:%M:%S')])
 
+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")