train.py

from processData import *
from diaryTraining import *
from diaryCandidateSet import multi_get_eachCityDiaryTop3000
from utils import get_date



# 把数据获取、特征转换、模型训练的模型串联在一起
if __name__ == "__main__":
    # while True:
    #     now = datetime.now()
    #     if (now.hour == 23) and (now.minute == 30):
    start_train = time.time()
    data_start_date, data_end_date, validation_date, test_date = get_date()
    data, test_number, validation_number = feature_en(data_start_date, data_end_date,
                                                      validation_date,test_date)
    ffm_transform(data, test_number, validation_number)
    train()
    end_train = time.time()
    print("训练模型耗时{}分".format((end_train-start_train)/60))
    print('---------------prepare candidates--------------')
    start = time.time()
    multi_get_eachCityDiaryTop3000()
    end = time.time()
    print("获取各城市热门日记耗时{}分".format((end - start) / 60))
    print("end")