CoolFace
Modelpublic

XaviArmengol/autotrain-titanic-54879128171

sourceHugging Faceupdated 3y agoView on Hugging Face
0likes32downloads
Model Card

Model Trained Using AutoTrain

  • —Problem type: Binary Classification
  • —Model ID: 54879128171
  • —CO2 Emissions (in grams): 0.8486

Validation Metrics

  • —Loss: 0.344
  • —Accuracy: 0.877
  • —Precision: 0.862
  • —Recall: 0.812
  • —AUC: 0.913
  • —F1: 0.836

Usage

python
import json
import joblib
import pandas as pd

model = joblib.load('model.joblib')
config = json.load(open('config.json'))

features = config['features']

# data = pd.read_csv("data.csv")
data = data[features]
data.columns = ["feat_" + str(col) for col in data.columns]

predictions = model.predict(data)  # or model.predict_proba(data)