我正在构建一个多标签文本分类程序,并试图使用OneVsRestClassifier+XGBClassifier对文本进行分类。最初,我使用Sklearn的Tf-以色列国防军矢量化文本,这没有错误。现在,我使用Gensim的Word2Vec来矢量化文本。然而,当我将矢量化数据输入OneVsRestClassifier+XGBClassifier时,在拆分测试和培训数据的行中会出现以下错误:
Type
我想使用我自己的word数据集来创建嵌入。并使用我自己的标签数据来训练和测试我的模型。为此,我已经使用word2vec创建了自己的单词嵌入。以及使用标签数据训练我的模型时遇到的问题。我在尝试训练模型时遇到错误。embeddings/read, embedding_1/Cast, embedding_1/GatherV2/axis)]]
Process finished wi