AssertionError: Массивы не равны - PullRequest
Новая
       24

AssertionError: Массивы не равны

0 голосов
/ 09 января 2020

Я занимаюсь разработкой нового классификатора и следую шаблону scikit-learn.

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

from sklearn.utils.estimator_checks import check_estimator
check_estimator(my_classifier)

Когда я запускаю приведенный выше код, я получаю сообщение об ошибке. Части ошибки ниже: 

~\Anaconda3\lib\site-packages\sklearn\utils\estimator_checks.py in check_classifiers_predictions(X, y, name, classifier_orig)
   1762         # For some specific cases 'ComplementNB' predicts less classes
   1763         # than expected
-> 1764         assert_array_equal(np.unique(y), np.unique(y_pred))
   1765     assert_array_equal(classes, classifier.classes_,
   1766                        err_msg="Unexpected classes_ attribute for %r: "

AssertionError: 
Arrays are not equal

(shapes (3,), (2,) mismatch)
 x: array(['one', 'three', 'two'], dtype='<U5')
 y: array(['one', 'three'], dtype='<U5')

Мой классификатор является двоичным классификатором. Кто-нибудь может указать мне направление на это? Большое спасибо.

...