train.py

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



# 把数据获取、特征转换、模型训练的模型串联在一起
if __name__ == "__main__":
    # while True:
    #     now = datetime.now()
    #     if (now.hour == 23) and (now.minute == 30):
    start = 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 = time.time()
    print("训练模型耗时{}分".format((end-start)/60))
    # print('---------------prepare candidates--------------')
    # get_eachCityDiaryTop3000()
    # print("end")