add cpickle

rov_train.py

@@ -152,6 +152,16 @@ def process_train_predict_data():
     train_data = getdatasample(train_day, 30, 'rov_feature_add_v1')
     predict_data = getdatasample(predict_day, 1, 'rov_predict_table_add_v1')
     #TODO save tempt
+    import _pickle as cPickle
+    with open('train_data.pickle','wb') as output_file:
+        cPickle.dump(train_data, output_file)
+    with open('predict_data.pickle','wb') as output_file:
+        cPickle.dump(predict_data, output_file) 
+    exit()
+    with open(r"train_data.pickle", "rb") as input_file:
+        train_data = cPickle.load(input_file)    
+    with open(r"predict_data.pickle", "rb") as input_file:
+        predict_data = cPickle.load(input_file)       
 
     train_data = basic_cal(train_data)
     predict_data = basic_cal(predict_data)
@@ -317,3 +327,6 @@ def do_train(train_data, predict_data, df_target, df_target_predict, df_new_feat
     sub_df_['score'] = predictions
     print('regre ranking shape', sub_df_.shape)
 
+
+if __name__ == '__main__':
+    process_train_predict_data()