chrstnv/titanic-survival

Titanic Survival Classifier

RandomForest (sklearn Pipeline: препроцессинг + классификатор в одном артефакте). Учебный проект.

Использование

import joblib, pandas as pd
from huggingface_hub import hf_hub_download

model = joblib.load(hf_hub_download("chrstnv/titanic-survival", "model.joblib"))
feats = ["Pclass", "Sex", "Age", "SibSp", "Parch", "Fare", "Embarked"]
model.predict(pd.read_csv("test.csv")[feats])   # принимает сырые данные

Данные

Обучена на chrstnv/titanic. Validation accuracy: 0.8156.