我正在使用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)