R 中的 Keras 和文本分类预测

Question

我正在关注这里的教程 ( https://tensorflow.rstudio.com/tutorials/beginners/basic-ml/tutorial_basic_text_classification/ )。 然后我喜欢对测试部分进行预测，所以这是我添加的代码：

# make predictions on testing
predictions <- model %>% predict(testing$text,epochs=20, batch_size=1,   validation_split = 0.1,
verbose = 0)
plot(predictions)
head(testing$text)
predictions

我现在想做的是将数值从predictions值转换为实际文本形式，例如示例中的“正”或“负”值，然后将这些值与testing$text合并以产生 dataframe。你能不能帮我弄清楚？

Answer 1

这样的事情应该有效：

results <- data.frame(
  text=testing$text, 
  class=ifelse(predictions>0.5,'positive','negative')
)

提示：如果您打算将 R 用于机器学习，您可能需要学习该语言的基础知识。