File size: 2,140 Bytes
64d77b1
1
{"model_id": "prabinpanta0/Patient-Readmission-Prediction", "model_type": "sequence-classification", "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": "Patient-Readmission-Prediction/random_forest_model.keras", "logistic_regression": "Patient-Readmission-Prediction/logistic_regression_model.keras", "k_nearest": "Patient-Readmission-Prediction/k_nearest_model.keras", "svc": "Patient-Readmission-Prediction/svc_model.keras", "naive_bayes": "Patient-Readmission-Prediction/naive_bayes_model.keras", "neural_network": "Patient-Readmission-Prediction/neural_network_model.keras", "cross_validation_random_forest": "Patient-Readmission-Prediction/cross_validation_random_forest_model.keras", "cross_validation_logistic_regression": "Patient-Readmission-Prediction/cross_validation_logistic_regression_model.keras", "cross_validation_lightgbm": "Patient-Readmission-Prediction/cross_validation_lightgbm_model.keras"}, "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.0, "max_iter": 1000}, "k_nearest": {"n_neighbors": 5}, "svc": {"C": 1.0, "kernel": "linear"}, "naive_bayes": {"alpha": 1.0}, "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.0, "max_iter": 1000}, "cross_validation_lightgbm": {"random_state": 42}}}