Update README.md

README.md

@@ -9,18 +9,107 @@ tags:
 - healthcare
 - sklearn
 - keras
-language: 
+- LightGBM
+language:
 - en
 model_type: classification
 task: patient-readmission-prediction
 app_mode: standalone
 frameworks:
-  - scikit-learn
-  - tensorflow
+- scikit-learn
+- tensorflow
 metrics:
-  - accuracy
-  - precision
-  - recall
-  - f1
+- accuracy
+- precision
+- recall
+- f1
+- roc_auc
 ---
-# Hospital Patient Readmission Predection
+# Hospital Patient Readmission Predection
+
+
+```json
+{
+  "model_id": "prabinpanta0/Patient-Readmission-Prediction",
+  "model_type": "sequence-classification",
+  "library": {
+    "random_forest": "scikit-learn",
+    "logistic_regression": "scikit-learn",
+    "k_nearest": "scikit-learn",
+    "svc": "scikit-learn",
+    "naive_bayes": "scikit-learn",
+    "neural_network": "keras",
+    "cross_validation_random_forest": "scikit-learn",
+    "cross_validation_logistic_regression": "scikit-learn",
+    "cross_validation_lightgbm": "LightGBM"
+  },
+  "model_architectures": {
+    "random_forest": "RandomForestClassifier",
+    "logistic_regression": "LogisticRegression",
+    "k_nearest": "KNeighborsClassifier",
+    "svc": "SVC",
+    "naive_bayes": "MultinomialNB",
+    "neural_network": "NeuralNetwork",
+    "cross_validation_random_forest": "RandomForestClassifier",
+    "cross_validation_logistic_regression": "LogisticRegression",
+    "cross_validation_lightgbm": "LGBMClassifier"
+  },
+  "model_paths": {
+    "random_forest": "model_RandomForestClassifier.pkl",
+    "logistic_regression": "model_Logistic_Regression.pkl",
+    "k_nearest": "model_K_nearest.pkl",
+    "svc": "model_svc.pkl",
+    "naive_bayes": "model_naive_bayes.pkl",
+    "neural_network": "neural_network.keras",
+    "cross_validation_random_forest": "model_rf.pkl",
+    "cross_validation_logistic_regression": "model_lr.pkl",
+    "cross_validation_lightgbm": "model_lgbm.pkl"
+  },
+  "model_classes": {
+    "random_forest": "RandomForestClassifier",
+    "logistic_regression": "LogisticRegression",
+    "k_nearest": "KNeighborsClassifier",
+    "svc": "SVC",
+    "naive_bayes": "MultinomialNB",
+    "neural_network": "NeuralNetwork",
+    "cross_validation_random_forest": "RandomForestClassifier",
+    "cross_validation_logistic_regression": "LogisticRegression"
+  },
+  "model_configs": {
+    "random_forest": {
+      "n_estimators": 100,
+      "max_depth": 5
+    },
+    "logistic_regression": {
+      "C": 1,
+      "max_iter": 1000
+    },
+    "k_nearest": {
+      "n_neighbors": 5
+    },
+    "svc": {
+      "C": 1,
+      "kernel": "linear"
+    },
+    "naive_bayes": {
+      "alpha": 1
+    },
+    "neural_network": {
+      "input_dim": 10,
+      "output_dim": 1,
+      "hidden_dim": 10
+    },
+    "cross_validation_random_forest": {
+      "n_estimators": 100,
+      "max_depth": 5
+    },
+    "cross_validation_logistic_regression": {
+      "C": 1,
+      "max_iter": 1000
+    },
+    "cross_validation_lightgbm": {
+      "random_state": 42
+    }
+  }
+}
+```