Passing list-likes to .loc or [] with any missing labels is no longer supported. The following labels were missing: Int64Index([...])

Ошибка:

Passing list-likes to .loc or [] with any missing labels is no longer supported. The following labels were missing: Int64Index([...])

Возникает после удаление строк из train (DataFrame)

train = train.drop(train[train['SalePrice'] > np.percentile(list(train['SalePrice']), 98)].index)

Далее, вроде делаю пересчет индексов, но не помогает

train.reset_index(drop=True)

kf = KFold(n_splits=5, shuffle=True)
kf.get_n_splits(train)

Возникает тут:

for i, (train_index, test_index) in enumerate(kf.split(train)):
    X_train, X_test = train.loc[train_index, selected_signs], train.loc[test_index, selected_signs]
    y_train, y_test = train.loc[train_index, 'SalePrice'], train.loc[test_index, 'SalePrice']