首页 > 解决方案 > 在 R keras 中,我想使用 predict_generator() 函数来获取 True y 值

问题描述

我想使用 R keras 中的 predict_generator 函数获得 True y 值,但我不知道如何。

但我能得到的是模型预测的y值。

我想得到生成器函数生成的测试数据的真实 y 值。

我需要帮助。

model <- keras_model_sequential() %>%
layer_gru(units = 32, dropout = 0.2, recurrent_dropout = 0.5,
input_shape = list(NULL, dim(data)[[-1]]-1)) %>%
layer_dense(units = 1, activation = 'sigmoid')

model %>% compile(
optimizer = optimizer_rmsprop(),
loss = "mse",
metrics = 'accuracy'
)  

history <- model %>% fit(
train_gen,
steps_per_epoch = 80,
epoch = 10,
validation_data = val_gen,
validation_steps = val_steps,

)

pred1 <- predict_generator(model, test_gen, steps=test_steps)
write.csv(pred1,"test(delay=48).csv")

标签: rtensorflowkerasgeneratorpredict

解决方案


您可以先使用生成测试数据,然后再应用预测。

test_data <- test_gen()
yhat <- predict(
   model,
   test_data[[1]],
   batch_size = batch_size,
   verbose = 0,
   steps = test_steps)

plot(yhat ~ test_data[[2]])

推荐阅读