add widget

README.md

@@ -1,14 +1,52 @@
 ---
 license: mit
----
-
----
 tags:
 - tabular-classification
 - sklearn
 datasets:
-- wine-quality
 - imodels/compas-recidivism
+widget:
+  structuredData:
+    age:
+      - 40.0
+    priors_count:
+      - 0.0
+    days_b_screening_arrest:
+      - -1.0
+    c_jail_time:
+      - 0.0
+    juv_fel_count:
+      - 0.0
+    juv_other_count:
+      - 0.0
+    juv_misd_count:
+      - 0.0
+    c_charge_degree:F:
+      - 0.0
+    c_charge_degree:M:
+      - 1.0
+    race:African-American:
+      - 0.0
+    race:Asian:
+      - 0.0
+    race:Caucasian:
+      - 1.0
+    race:Hispanic:
+      - 0.0
+    race:Native_American:
+      - 0.0
+    race:Other:
+      - 0.0
+    age_cat:25_-_45:
+      - 1.0
+    age_cat:Greater_than_45:
+      - 0.0
+    age_cat:Less_than_25:
+      - 0.0
+    sex:Female:
+      - 0.0
+    sex:Male:
+      - 1.0
 ---
 
 
@@ -31,14 +69,6 @@ X_test = df.drop(columns=['is_recid'])
 y_test = df['is_recid'].values
 ```
 
-### Load the model
-## Wine Quality classification
-
-### A Simple Example of Scikit-learn Pipeline
-
-> Inspired by https://towardsdatascience.com/a-simple-example-of-pipeline-in-machine-learning-with-scikit-learn-e726ffbb6976 by Saptashwa Bhattacharyya
-
-
 ### Load the model
 
 ```python
@@ -61,4 +91,5 @@ model = joblib.load(cached_download(
 ```
 preds = model.predict(X_test)
 print('accuracy', np.mean(preds==y_test))
+# accuracy 0.6759165485112416
 ```