как мы можем иметь дело с меткой, включающей 3 порядковых значения в машинном обучении

Question

Меня интересует создание модели прогнозирования, которая зависит от порядковых категорий, у меня есть несколько столбцов, в которых содержится большая часть уровней серьезности Высокий, Средний, низкий, и у меня есть метка общей серьезности, могу ли я создать хорошую модель для прогнозированияобщая степень тяжести с помощью машинного обучения?

Благодарим Вас за любые ответы, большое спасибо!

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]))