我正在研究文本分类,这里有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'))),
])
在此之后,我已经适应管道并将分类器存储在pickle中
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)