train.py

from processData import *
from diaryTraining import *
from diaryCandidateSet import get_eachCityDiaryTop3000


# 把数据获取、特征转换、模型训练的模型串联在一起
if __name__ == "__main__":
    data, test_number, validation_number = feature_en()
    ffm_transform(data, test_number, validation_number)
    train()

    print('---------------prepare candidates--------------')
    get_eachCityDiaryTop3000()
    print("end")