Downloads · 30 days
5
2% of all-time downloads
autotrain-titanic-744222727
vabadeh213/autotrain-titanic-744222727
autotrain-titanic-744222727 is a machine learning model from vabadeh213. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
https://colab.research.google.com/drive/16rmsJTBelh2vIWVxt9ncFEJmU7cEdUsE?usp=sharing
All-time downloads
229
Public
Repo size
299 KB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.joblib299 KB · 99%
At a glance
- Library
- transformers
- Model type
- xgboost
- Access
- Public
- Created
- Apr 15, 2022
- Updated
- Apr 15, 2022
- SHA
- f2fe8e0b
Model card
From the Hugging Face model README
https://colab.research.google.com/drive/16rmsJTBelh2vIWVxt9ncFEJmU7cEdUsE?usp=sharing
Model Trained Using AutoTrain
- Problem type: Binary Classification
- Model ID: 744222727
- CO2 Emissions (in grams): 0.00509303545772981
Validation Metrics
- Loss: 0.40596098709549455
- Accuracy: 0.8378378378378378
- Precision: 0.8518518518518519
- Recall: 0.92
- AUC: 0.8866666666666667
- F1: 0.8846153846153846
Usage
import json
import joblib
model = joblib.load('model.joblib')
config = json.load(open('config.json'))
features = config['features']
# data = pd.read_csv("data.csv")
data = data[features]
predictions = model.predict(data) # or model.predict_proba(data)
- Library
- transformers
- Type
- xgboost
- Created
- Apr 15, 2022
- Updated
- Apr 15, 2022