Python ForestNet示例

示例#1

文件： gbdt_lstm_ecg.py 项目： guohuiGH/graduation_design

def main():
    dataset = InputLibsvm()
    #print (dataset.x_train.shape)
    print (dataset.x_train[0].shape)
    print (dataset.x_test[0].shape)
    param = LstmParam(dataset)
    model = LstmModel(param)
    forest_net = ForestNet(param.batch_size, param.f_number, param.f_learning_rate, param.f_steps, param.f_input, param.training_iters, param.thread_core)
    train(param, model, forest_net, dataset)

示例#2

文件： gbdt_lstm_mnist.py 项目： guohuiGH/graduation_design

def main():
    param = LstmParam()
    model = LstmModel(param)
    forest_net = ForestNet(param.batch_size, param.f_number, param.f_learning_rate, param.f_steps, param.f_input, param.training_iters)
    train(param, model, forest_net)