【发布时间】:2018-09-14 18:13:18
【问题描述】:
这是我的 SVC 分类器代码。
vectorizer = TfidfVectorizer(lowercase=False)
train_vectors = vectorizer.fit_transform(training_data)
classifier_linear = svm.LinearSVC()
clf = CalibratedClassifierCV(classifier_linear)
linear_svc_model = clf.fit(train_vectors, train_labels)
training_data 这里是英文句子列表,train_lables 是相关的标签。在创建training_data 的最终版本之前,我会进行通常的停用词删除和一些预处理。这是我的测试代码:
test_lables = ["no"]
test_vectors = vectorizer.transform(test_lables)
prediction_linear = clf.predict_proba(test_vectors)
counter = 0
class_probability = {}
lables = []
for item in train_labels:
if item in lables:
continue
else:
lables.append(item)
for val in np.nditer(prediction_linear):
new_val = val.item(0)
class_probability[lables[counter]] = new_val
counter = counter + 1
sorted_class_probability = sorted(class_probability.items(), key=operator.itemgetter(1), reverse=True)
print(sorted_class_probability)
现在,当我使用训练集中已经存在的短语(在本例中为单词“no”)运行代码时,它可以正确识别,但置信度分数甚至低于 .9。输出如下:
[('no', 0.8474342514152964), ('hi', 0.06830103628879058), ('thanks', 0.03070201906552546), ('confused', 0.02647134535600733), ('ok', 0.015857384248465656), ('yes', 0.005961945963546264), ('bye', 0.005272017662368208)]
当我在线学习时,我发现训练集中已经存在的数据的置信度得分通常更接近1 或几乎1,其余的都可以忽略不计。我该怎么做才能获得更好的置信度分数?我是否应该担心,如果我添加更多的类,置信度会进一步下降,并且我很难确定一个突出的类?
【问题讨论】:
-
如果您要投反对票,请留言说明原因。射击和滑行对任何人都没有帮助。
标签: machine-learning scikit-learn svm