仅显示来自 classifier.predict_proba(X) 的前 n 个结果
Display only the top n results from classifier.predict_proba(X)
我可以让 sklearn 的 classifier.predict_proba(X) 只显示高于 m% 的前 n 个概率，而不是显示每个 class 的概率吗？是否有 option/parameter 允许我这样做？
没有参数可以做到这一点，但你可以简单地做一个 argsort:
np.argsort(classifier.predict_proba(X), axis=1)[-5:]
我可以让 sklearn 的 classifier.predict_proba(X) 只显示高于 m% 的前 n 个概率，而不是显示每个 class 的概率吗？是否有 option/parameter 允许我这样做？
没有参数可以做到这一点，但你可以简单地做一个 argsort:
np.argsort(classifier.predict_proba(X), axis=1)[-5:]