change main for test

d96da3e4 · 张彦钊 · 751ba854 · d96da3e4 · d96da3e4
Commit d96da3e4 authored Aug 10, 2018 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 5 deletions

processData.py processData.py +0 -3

train.py train.py +2 -2

No files found.
--- a/processData.py
+++ b/processData.py
@@ -106,8 +106,5 @@ def ffm_transform(data, test_number, validation_number):
    train.to_csv(DIRECTORY_PATH + "train{0}-{1}.csv".format(DATA_START_DATE, VALIDATION_DATE), index=False, header=None)


-if __name__ == "__main__":
-    data_fe = feature_en()
-    ffm_transform(data_fe)


--- a/train.py
+++ b/train.py
@@ -8,6 +8,6 @@ if __name__ == "__main__":
    data, test_number, validation_number = feature_en()
    ffm_transform(data, test_number, validation_number)
    train()
-    # print('---------------prepare candidates--------------')
-    # get_eachCityDiaryTop3000()
+    print('---------------prepare candidates--------------')
+    get_eachCityDiaryTop3000()