Commit From AutoTrain

.gitattributes

@@ -32,3 +32,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.bin.* filter=lfs diff=lfs merge=lfs -text
+*.tar.gz filter=lfs diff=lfs merge=lfs -text
+tokenizer.json filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,46 @@
+---
+tags:
+- autotrain
+- tabular
+- classification
+- tabular-classification
+datasets:
+- desertdev/autotrain-data-imdb-sentiment-analysis
+co2_eq_emissions:
+  emissions: 0.5768634462494043
+---
+
+# Model Trained Using AutoTrain
+
+- Problem type: Binary Classification
+- Model ID: 44994113085
+- CO2 Emissions (in grams): 0.5769
+
+## Validation Metrics
+
+- Loss: 0.685
+- Accuracy: 0.565
+- Precision: 0.565
+- Recall: 1.000
+- AUC: 0.500
+- F1: 0.722
+
+## 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)
+
+```

config.json

@@ -0,0 +1 @@
+{"features": ["text"], "targets": ["target"], "model_type": "xgboost", "target_mapping": {"hate_speech": 0, "no_hate_speech": 1}}

model.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2427ae334562591375738d2e94cf24550f3a64ada1ffe71d0f70eea39d72a59e
+size 223867