jamesmullenbach / caml-mimic

multilabel classification of EHR notes
MIT License
278 stars 125 forks source link

Parameter Y (size of label space) is overwritten #19

Open aplz opened 5 years ago

aplz commented 5 years ago

This is just a minor issue but might be helpful for other users. The parameter Y of the training script is not used for building the model. Instead, the size of the label space is set in the method 'pick_model' from the dictionary computed during data processing. This makes sense but then the parameter Y should be removed.