R中在keras中使用layer_dense进行神经网络配置

时间:2019-07-19 22:35:21

标签: r keras

我正在尝试使用keras在R中训练ANN。我有一个165个变量的输入,一个只有一个变量的输出。我的样本集包含多个观察值。假设ANN包含一个输入层,一个或多个隐藏层以及一个输出层。然后,每个隐藏层都可以由layer_dense定义。第一个隐藏层需要“ input_shape”参数,该参数等于输入变量的数量。如果我的理解是正确的,那么在我的示例中,“ units”参数应该为1,以表示单个输出变量。底部提供了模型定义代码。

但是,如果我将“单位”设置为1,则acc和val_acc为0;如果将其更改为2或更大,则得到正acc和val_acc。但这对我来说毫无意义。

然后,我使用以下代码进行预测。对于“单位” = 1,y_predict只有一列;对于“单位” = 2,y_predict有两列,我不知道该如何处理。

y_predict <- model %>% predict(x_test, batch_size = 128)

有人可以使用layer_sense给我更多有关NN配置的解释吗?非常感谢。

model = keras_model_sequential()

model %>% layer_dense(units = 10, activation = 'relu', input_shape = 165)

model %>% layer_dense(units = 6, activation = 'relu')

model %>% layer_dense(units = 1, activation = 'relu')

0 个答案:

没有答案