update

2 files changed, 2 insertions, 0 deletions

diff --git a/seer.py b/seer.py
index 006eaf3..9226c39 100755
--- a/seer.py
+++ b/seer.py
@@ -28,6 +28,7 @@ def launch_ais(which):
     elif which == "lstm_type_2": lstm_type_2()
     elif which == "lstm_type_3": lstm_type_3()
     elif which == "cnn_type_1": pass
+    elif which == "tfann_type_1": tfann_type_1()
     else: pass
 
 # write code here
diff --git a/tfann.py b/tfann.py
index 326182c..aef226a 100755
--- a/tfann.py
+++ b/tfann.py
@@ -35,6 +35,7 @@ def GetCurDF(cur, fp):
     openUrl.close()
     df = pd.read_json(r.decode())
     df['date'] = df['date'].astype(np.int64) // 1000000000
+    print(df.head())
     return df
 
 class PastSampler: