如何使用Glove字嵌入构建模型,并使用R中的text2vec预测测试数据

我正在使用GloVe字嵌入将文本数据分类模型分为两类(即将每个评论分为两类).我有两列,一列是文本数据(注释),另一列是二元Target变量(注释是否可操作).我能够使用text2vec文档中的以下代码为文本数据生成Glove字嵌入.

glove_model <- GlobalVectors$new(word_vectors_size = 50,vocabulary = 
glove_pruned_vocab,x_max = 20L)
#fit model and get word vectors
word_vectors_main <- glove_model$fit_transform(glove_tcm,n_iter = 20,convergence_tol=-1)
word_vectors_context <- glove_model$components
word_vectors <- word_vectors_main+t(word_vectors_context)

如何构建模型并生成测试数据的预测?

最佳答案 text2vec有一个标准的预测方法(就像大多数R库一样),你可以直接使用:看一下
documentation.

总而言之,只需使用即可

predictions <- predict(fitted_model, data)
点赞