Я столкнулся с этой ошибкой с алгоритмом Naive Bayes's CategoricalNB
Он дает указанную выше ошибку после 2-й попытки Я запускаю ячейки. Это означает, что он работает без ошибок в первый раз, и когда я пытаюсь что-то изменить (например, комментарий) и снова запустить ноутбук, он выдает ошибку:
IndexError: индекс 38 равен за пределами оси 1 размером 38
Я не знаю, что не так и как исправить. Когда я перезапускаю ядро и пробую снова, оно работает, и каждая попытка после 1-й попытки терпит неудачу и дает указанную выше ошибку.
%matplotlib inline
import matplotlib.pyplot as plt
import pandas as pd
dataframe = pd.read_csv("hr_dataset.csv")
# dataframe = pd.read_csv("WA_Fn-UseC_-HR-Employee-Attrition.csv")
dataframe.head(2)
from sklearn.naive_bayes import CategoricalNB
from sklearn.metrics import accuracy_score
from sklearn.model_selection import train_test_split
# inputs = scaled_df
X_train, X_test, y_train, y_test = train_test_split(inputs, target, test_size=0.2)
categoricalNB_ = CategoricalNB()
categoricalNB_.fit(X_train, y_train)
X_train.shape, X_test.shape, y_train.shape, y_test.shape
pred = categoricalNB_.predict(X_test) # --------------> gives the error for every attempt after the 1st attempt. weird
categoricalNB_.score(X_test, y_test)
# accuracy_score(y_test,pred)