Deepnet软件包输出所需范围之外的概率

时间:2017-01-23 12:56:41

标签: r neural-network

我使用deepnet进行分类。数据集中的所有内容都是数字。我使用以下命令训练我的模型:

library(deepnet)
dnn <- dbn.dnn.train(training, y_training, hidden = rep(100,50,30),numepochs = 5, learningrate = 0.5, hidden_dropout = 0.3)

prednn <- nn.predict(dnn, testing)
range(prednn)
[1] 0.7167986 0.9834596

一切都很好,但是当我预测使用deepnet时,我会在上述范围内获得输出。我不应该在0和1之间得到输出,我可以基于诸如0.5的截止值进行分割。或者我在模型中没有指定任何东西。

更新 使用非常低的学习率我会得到更糟糕的结果。

dnn <- dbn.dnn.train(training, y_training, hidden = rep(100,50,30),numepochs = 5, learningrate = 0.1, hidden_dropout = 0.3)

 range(prednn)
[1] 0.8619268 0.8619268

0 个答案:

没有答案