我想知道是否要创建一个基于顺序类别的预测模型,我有多个列,其中大多数包含严重性级别,高,中,低,并且我的标签是总体严重性,是否可以创建一个好的模型来进行预测使用机器学习的整体严重程度?
非常感谢您的任何答复!
我想知道是否要创建一个基于顺序类别的预测模型,我有多个列,其中大多数包含严重性级别,高,中,低,并且我的标签是总体严重性,是否可以创建一个好的模型来进行预测使用机器学习的整体严重程度?
非常感谢您的任何答复!
您可以使用如下所示的序数回归:
$ 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]))