Задача курса
predictproba и classes: найдите вероятность нужного класса
Курс «Машинное обучение с нуля на Python: первая модель» · урок «predict_proba и порядок столбцов»
Условие
Вероятность класса 2 для каждой строки
Для новых доставок нужно получить вероятность именно интересующего исхода с меткой 2 и расположить строки от большего риска к меньшему. Модель различает метки 2 и 5; положение нужного столбца определяется clf.classes_.
Что уже дано
Видимый код создаёт числовые X_train, y_train, обученный clf и четыре новые строки X_new с индексами A–D. Скрытая подготовка не создаёт данные. clf.predict_proba(X_new) возвращает столбцы в порядке clf.classes_.
Что нужно сделать
Реализуйте rank_class_two(clf, rows). Найдите в clf.classes_ метку 2, извлеките её вероятности и верните DataFrame с единственным столбцом risk_class_2. Индекс должен сохранить идентификаторы входных строк; строки отсортируйте по вероятности убыванию, при равенстве сохраните исходный порядок. Сохраните вызов функции для X_new в ranking.
Ввод и вывод
stdin не используется. Печатать через print в stdout ничего не нужно. Проверяется ranking; затем скрытая проверка вызывает rank_class_two для других строк и другого порядка классов. Возвращаемое значение — pandas DataFrame с вероятностями прогноза, а не известными исходами.
