Commit Live PR

q01_load_data/build.py

@@ -1,7 +1,15 @@
+# %load q01_load_data/build.py
 import pandas as pd
 import numpy as np
 from sklearn.model_selection import train_test_split
 
-"write your solution here"
-
+'write your solution here'
+def q01_load_data(path):
+    df = pd.read_csv(path)
+    return df
+q01_load_data('data/perrin-freres-monthly-champagne.csv')
+
+
+
+
 

q02_data_splitter/build.py

@@ -1,3 +1,4 @@
+# %load q02_data_splitter/build.py
 import pandas as pd
 import numpy as np
 import datetime
@@ -6,4 +7,15 @@
 #sys.path.append('./')
 from greyatomlib.time_series_101_project.q01_load_data.build import q01_load_data
 
-'''Write your solution here'''
+#'''Write your solution here''
+
+def q02_data_splitter(path):
+    tss=q01_load_data(path)
+    tss['Month']=pd.to_datetime(tss['Month'])
+    X_train = tss[tss['Month'] <datetime.datetime(1971, 10, 1, 0, 0, 0)]
+    X_valid = tss[tss['Month'] >=datetime.datetime(1971, 10, 1, 0, 0, 0)]
+    return X_train, X_valid
+
+
+
+