Hugging Face's logo

Sumitkumar098
/
BiomedNLP-HealthPredict

Model card Files
xet
Community
Sumitkumar098 commited on
Commit
d186cf8
·
verified ·
1 Parent(s): 836c3b3

Main Model

Browse files
Files changed (4) hide show
  1. label_encoders.pkl +3 -0
  2. lookup_data.pkl +3 -0
  3. model_config.json +1 -0
  4. model_state_dict.pt +3 -0
label_encoders.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f3a1af741990f399ebc937603adeb05c2e4b80a61700edccbc0f778826dafeba
3
+ size 1037
lookup_data.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a4a69ec521ddb95f94a83fd00cbc0600ae68d9a9b00e86ec629ed187fc1d7d2
3
+ size 3037
model_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"model_name": "microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract", "num_medications": 27, "num_polypharmacy_classes": 3, "num_disease_classes": 5}
model_state_dict.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad0502269446dc83471d34713e58b0fca401cd7ff90ce4326bc4b1b40aac64b5
3
+ size 443976809