Байесовская сеть в гранате: ValueError: Образец не имеет того же числа измерений, что и модель

Question

Я пытаюсь смоделировать байесовскую сеть в python, используя пакет граната. Сеть должна быть извлечена из данных. Поэтому я использую метод.from_samples. Однако у меня возникают проблемы с использованием метода.predict_proba(), и это дает мне ошибку.

Вот как я строю модель:

model = BayesianNetwork.from_samples(X_train, algorithm='chow-liu')

и вот как я делаю прогноз:

model.predict_proba(X_train)

и это ошибка, которую я получаю:

ValueError: Sample не имеет того же числа измерений, что и модель. Ваша помощь будет высоко ценится.

Answer 1

Добавьте [] вокруг образца, который вы передаете

Answer 2

Я получил ответ: вы должны определить ваши имена_состояний при вызове метода from_samples.

Другой вопрос, как мы делаем классификацию, используя эту модель?

Answer 3

Вы должны использовать метод предиката () для прогнозирования состояния не оцениваемых узлов.

Проверьте документацию для более подробной информации. Также в репозитории вы можете найти несколько интересных уроков, которые помогут вам.