python - 我们如何处理标签在机器学习中包含 3 个序数值

Question

我想知道创建一个依赖于序数类别的预测模型，我有多个列，其中包含大部分严重级别，高、中、低，我的标签是总体严重性，我可以创建一个好的模型来预测总体严重性使用机器学习？

非常感谢任何回应！

标签： pythonrmachine-learning

Answer 1

您可以使用如下的序数回归：

$ pip install mord

from mord import LogisticAT
import numpy as np

replace_dict = {'High':3,'Medium':2,'Low':1}

y_values = ['High','Medium','Low']

y = [replace_dict[i] for i in y_values]


X = np.array([[1,2,3],[3,4,5],[6,7,8]])
y = np.array([y])

ord = LogisticAT()
ord.fit(X,y) 
ord.predict(np.array([10,11,12]))