我使用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