Commit d96da3e4 authored by 张彦钊's avatar 张彦钊

change main for test

parent 751ba854
...@@ -106,8 +106,5 @@ def ffm_transform(data, test_number, validation_number): ...@@ -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) 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)
...@@ -8,6 +8,6 @@ if __name__ == "__main__": ...@@ -8,6 +8,6 @@ if __name__ == "__main__":
data, test_number, validation_number = feature_en() data, test_number, validation_number = feature_en()
ffm_transform(data, test_number, validation_number) ffm_transform(data, test_number, validation_number)
train() train()
# print('---------------prepare candidates--------------') print('---------------prepare candidates--------------')
# get_eachCityDiaryTop3000() get_eachCityDiaryTop3000()
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment