【问题标题】:ValueError: dimension mismatch tfidf naive bayesValueError:尺寸不匹配 tfidf 朴素贝叶斯
【发布时间】:2019-03-29 22:49:34
【问题描述】:

我真的需要你的帮助。我正在研究文本分类,我使用了tfidfnaive bayes ngram,所以我保存了我的模型,然后我加载它以重用和分类一个短语,问题是在我使用加载的tfidf 之后,当我尝试使用我的模型我收到了这个错误

"ValueError: 尺寸不匹配" 提前致谢

from Feature_Engineering import clean_text_test
import pandas as pd 
X_raw = 'Spécialisées dans l’achat et la revente de produits de luxes, ces organisations parviennent à blanchir des centaines de millions d’euros par an. Le luxe à la française fascine les jeunes Chinois de la classe moyenne. C’est tellement vrai qu’en région parisienne des organisations occultes profitent de la revente de vêtements et d’articles de maroquinerie pour blanchir de l’argent provenant de fraudes fiscales, du proxénétisme ou du trafic de drogue. Les enquêteurs du deuxième district de police judiciaire de Paris viennent de mettre à bas, à l’automne dernier, un réseau fort d’au moins six hommes et femmes d’origine chinoise, âgés de 30 et 40 ans, qui entre 2017 et 2018, a fait circuler plus de cinq millions d’euros entre la France et la Chine, sans payer un centime de taxe.'

X_final=clean_text_test(X_raw)

df_train = pd.DataFrame({'X': [X_final]})

tf1 = pickle.load(open("tf_ngram.pkl", 'rb'))

# Create new tfidfVectorizer with old vocabulary
tf1_new = TfidfVectorizer(analyzer='word', ngram_range=(5,5),  lowercase = True,
                          max_features = 50000, vocabulary = tf1.vocabulary_)
X_tf1 = tf1_new.fit_transform(df_train.X)

filename = 'Naive_Bayes_CountVect'

loaded_model = pickle.load(open(filename, 'rb'))

ypredict=loaded_model.predict(X_tf1)

【问题讨论】:

    标签: python


    【解决方案1】:

    为什么要再次安装 TF-IDF 矢量化器?你只需要做X_tf1 = tf1.transform(df_train.X)

    【讨论】:

    • 我使用了 X_tf1 = tf1.transform(df_train.X) 但我仍然有同样的错误
    猜你喜欢
    • 2015-10-28
    • 2017-08-27
    • 2021-06-01
    • 2012-02-21
    • 2011-12-28
    • 2012-11-28
    • 2013-09-09
    • 2015-01-03
    • 2017-02-09
    相关资源
    最近更新 更多