KaiquanMah commited on
Commit
ceabc60
·
verified ·
1 Parent(s): f8b9a00

Update preprocess.py

Browse files
Files changed (1) hide show
  1. preprocess.py +1 -0
preprocess.py CHANGED
@@ -26,6 +26,7 @@ def parse(csv_path):
26
 
27
  # Extract datetime features
28
  X.loc[:,'DateTime'] = pd.to_datetime(X['DateTime'], errors='coerce')
 
29
  X.loc[:,'DayOfWeek'] = X['DateTime'].dt.dayofweek
30
  X.loc[:,'Month'] = X['DateTime'].dt.month
31
  X.loc[:,'Hour'] = X['DateTime'].dt.hour
 
26
 
27
  # Extract datetime features
28
  X.loc[:,'DateTime'] = pd.to_datetime(X['DateTime'], errors='coerce')
29
+ X = X.dropna(subset=['DateTime'])
30
  X.loc[:,'DayOfWeek'] = X['DateTime'].dt.dayofweek
31
  X.loc[:,'Month'] = X['DateTime'].dt.month
32
  X.loc[:,'Hour'] = X['DateTime'].dt.hour