Merge branch 'main' of https://sc.thorchat.ru/masha/Valar-Morgulis
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
# Прогноз смертности персонажей «Песни льда и огня»
|
||||
|
||||
Учебный проект по бинарной классификации: модель оценивает вероятность смерти персонажа на основе его возраста, пола, популярности, принадлежности к дому, культуры и других характеристик.
|
||||
Модель оценивает вероятность смерти персонажа на основе его возраста, пола, популярности, принадлежности к дому, культуры и других характеристик.
|
||||
|
||||
## Структура проекта
|
||||
|
||||
@@ -21,7 +21,7 @@
|
||||
|
||||
Лучший ROC AUC показал Random Forest — около **0.74**. Он также получил F1 около **0.49**. Самым важным признаком оказалась популярность персонажа, заметно меньший вклад внесли возраст, пол и количество погибших родственников.
|
||||
|
||||
Dummy-модель имеет высокую accuracy из-за преобладания живых персонажей, но не обнаруживает ни одной смерти. Это показывает, почему при несбалансированных классах важно учитывать не только accuracy, но также precision, recall, F1 и ROC AUC.
|
||||
Dummy-модель имеет высокую accuracy из-за преобладания живых персонажей, но не обнаруживает ни одной смерти.
|
||||
|
||||
## Запуск
|
||||
|
||||
|
||||
Reference in New Issue
Block a user