39のカテゴリ/クラスと850万件のレコードがあるテキスト分類に取り組んでいます。(将来的にはデータとカテゴリーが増えるでしょう)。
私のデータの構造またはフォーマットは次のとおりです。
----------------------------------------------------------------------------------------
| product_title | Key_value_pairs | taxonomy_id |
----------------------------------------------------------------------------------------
Samsung S7 Edge | Color:black,Display Size:5.5 inch,Internal | 211
Storage:128 GB, RAM:4 GB,Primary Camera:12 MP
Case cover Honor 8 | Color:transparent,Height:15 mm,width:22 mm | 212
Ruggers Men's T-Shirt | Size:L,ideal for:men,fit:regular, | 111
sleeve:half sleeve
Optimum Nutrition Gold | Flavor:chocolate,form:powder,size:34 gm | 311
Standard Whey Protein
データの分布は正常ではありません。それは非常に不均衡です:
-------------------------
| taxonomy_id | count |
-------------------------
111 | 851750
112 | 355592
113 | 379433
114 | 23138
115 | 117735
116 | 145757
117 | 1339471
121 | 394026
122 | 193433
123 | 78299
124 | 111962
131 | 1776
132 | 4425
133 | 908
134 | 23062
141 | 22713
142 | 42073
211 | 7892
212 | 1574744
221 | 1047
222 | 397515
223 | 53009
231 | 1227
232 | 7683
251 | 739
252 | 327
253 | 38974
254 | 25
311 | 2901
321 | 7126
412 | 856
421 | 697802
422 | 414855
423 | 17750
425 | 1240
427 | 658
429 | 1058
431 | 20760
441 | 257
ご覧のとおり、それらは非常に不均衡であり、誤分類につながっています。
これまでに実行した手順
1)product_titleとkey_value_pairs列をマージし、ストップワードと特殊文字を削除してステミングを実行します。
2)TFIDFvectorizer()、LinearSVC()にパイプラインを使用しました
vectorizerPipe = Pipeline([
('tfidf', TfidfVectorizer(lowercase=True, stop_words='english')),
('classification', OneVsRestClassifier(LinearSVC(penalty='l2', loss='hinge'))),
])
この後、パイプラインを適合させ、分類器をピクルスに格納しました
prd = vectorizerPipe.fit(df.loc[:, 'description'], df.loc[:, 'taxonomy_id'])
テスト側では、上記のようにステップ1を繰り返し、次にピクルスをロードして予測関数を使用しました
pd = cl.predict([testData])
私が直面している問題
多くの製品が他のいくつかのカテゴリに誤って分類されています
例:Ultimate Nutrition Prostar 100%ホエイプロテインはカテゴリ311に分類する必要がありますが、私の分類子はそれを222として分類していますが、これは完全に間違っています。
TFidfVectorizer()とHashingvectorizer()のどちらを使用するかわからないのですが、パラメーターと一緒にこれを選択するのを手伝っていただけますか?
私が使用しているアルゴリズムはLinearSVCですが、大量のデータを含むマルチクラス分類問題に適していますか?または、別のアルゴリズムを使用する必要がありますか?
私のデータは非常に不均衡なので、ランダムアンダーサンプリングを試しました。結果は改善されましたが、まだ基準に達していませんでした。また、これがランダムアンダーサンプリングを実行する適切なアプローチであるかどうかもわかりません。
pipe = make_pipeline_imb( HashingVectorizer(lowercase=True), RandomUnderSampler(ratio={111: 405805, 112: 170431, 113: 241709, 114: 8341, 115: 50328, 116: 89445, 117: 650020, 121: 320803, 122: 162557, 123: 66156, 124: 36276, 131: 1196, 132: 3365, 133: 818, 134: 15001, 141: 6145, 142: 31783, 211: 24728, 212: 100000, 221: 791, 222: 8000, 223: 35406, 231: 785, 232: 3000, 251: 477, 252: 127, 253: 29563, 254: 33, 311: 2072, 321: 5370, 412: 652, 421: 520973, 422: 99171, 423: 16786, 425: 730, 427: 198, 429: 1249, 431: 13793, 441: 160},random_state=1), OneVsRestClassifier(LinearSVC(penalty='l2', loss='hinge')))
私は機械学習の初心者なので、この方法をテキスト分類に使用しました。私のアプローチが間違っている場合は、正しい方法で修正してください。
(私がよりよく理解するのに役立つので、例を挙げて提案や解決策を提供してくれると素晴らしいでしょう)
***編集-1 ****
RndmFrst = RandomForestClassifier(n_estimators=100, max_depth=20, max_features=5000,n_jobs=-1)
LogReg = LogisticRegression()
voting = VotingClassifier(estimators=[('LogReg ', LogReg), ('RndmFrst', RndmFrst)], voting='soft', n_jobs=-1)
pipe = Pipeline([('tfidf', TfidfVectorizer(ngram_range=(1,4), max_features=50000)), ('clf', voting)])
pipe = pipe.fit(df.loc[:,'description'], df.loc[:,'taxonomy_id'])
Preds = pipe.predict(test_data)