NeuralNet nolearn的一致性测试结果

时间:2016-02-05 03:30:03

标签: python nolearn

我在nolearn库中使用NeuralNet类来做分类任务。这是代码:

layers0 = [('input', InputLayer),
           ('hidden', DenseLayer),
           ('output', DenseLayer)]

net0 = NeuralNet(layers=layers0,
                 input_shape=(None, 7),
                 hidden_num_units=7,
                 output_num_units=6,
                 output_nonlinearity=softmax,
                 update=nesterov_momentum,
                 update_learning_rate=0.1,
                 update_momentum=0.2,
                 train_split=TrainSplit(eval_size=0),
                 verbose=0,
                 max_epochs=200)
                 net0.fit(X, y)

predict = net0.predict(X_test)
print confusion_matrix(ids, predict)
print "accuracy: ", accuracy_score(ids, predict)

此代码训练NeuralNet并在测试集上进行测试。但是当我多次运行时,每个都会产生不同的结果。那么如何根据参数,训练集和测试集训练NeuralNet只给出一个结果?

1 个答案:

答案 0 :(得分:0)

在调用net0.fit()之前,只需使用种子到随机数生成器。例如......

numpy.random.seed(123)