Exemple #1
0
def run():
    
    print('\n加载训练数据')
    train_ds = DataSource.DataSource(train_data_index,train_data_zip);
    model = BPModel(feature_size,label_size,hidden_size,act_func,rou);
    if need_train:
        print('\n开始训练')
        model.train(train_ds, learn_rate, steps, batch_size,load_values);
    print('\n开始测试')
    train_ds.reload_index(test_data_index);
    print('开始测试输出')
    _,py,y=model.calculate(train_ds);
    print('开始评价')
    evel(py,y);
    
    if need_result_out:
        print('\n开始加载比赛数据集')
        result_ds=DataSource.DataSource(result_test_index,result_test_data_zip);
        indexids=result_ds.getAllIndexID();
        print('计算比赛数据')
        py=model.cal_py_out(result_ds);
        print('创建比赛文件')
        DataSetProcess.create_result_csv(indexids, py, result_test_out_path);

    pass;
def run():

    print('\n加载训练数据')
    train_ds = DataSource.DataSource(train_data_index, train_data_zip)
    if need_train:
        print('\n开始训练')
        train_ae(train_ds, )


#     print('\n开始测试')
#     train_ds.reload_index(test_data_index);
#     print('开始测评')
#     evel(train_ds);

    if need_result_out:
        print('\n开始加载比赛数据集')
        result_ds = DataSource.DataSource(result_test_index,
                                          result_test_data_zip)
        indexids = result_ds.getAllIndexID()
        print('计算比赛数据')
        py = calculate(result_ds)
        print('创建比赛文件')
        DataSetProcess.create_result_csv(indexids, py, result_test_out_path)

    pass