Skip to content

Releases: Saiful185/AMR-EnsembleNet

AMR Ensemble Pretrained Base Models

26 Sep 16:41
54077c3

Choose a tag to compare

This release contains the pre-trained AMR model weights for the final Ensemble. Feel free to use them for demonstration/production.

  • AMR_cnn1d_model_[Antibiotic_Name].keras: The custom 1D CNN model weights trained on the Giessen dataset for specific antibiotics.
  • AMR_xgboost_model_[Antibiotic_Name].json: The XGBoost models trained on the Giessen dataset for specific antibiotics.
  • AMR_RF_model_[Antibiotic_Name].pkl: The Random Forest models trained on the Giessen dataset for specific antibiotics.