Skip to content

Latest commit

 

History

History
 
 

baseline

NN Baseline Model

A neural network (not Recurrent) with one hidden layer, with relu activation on the hidden layer and sigmoid activation on the output layer. Using cross entropy loss,which is the loss functions for multilabel classification [4]

Evaluation

rank loss metric to evaluate performance [3] and F1 score [2]

References

[1] Diagnosis code assignment: models and evaluation metrics. Journal of the American Medical Informatics
[2] Applying Deep Learning to ICD-9 Multi-label Classification from Medical Records
[3] ICD-9 Coding of Discharge Summaries
[4] Large-scale Multi-label Text Classification - Revisiting Neural Networks