Skip to content

Multiclass classification of doctors from text/keywords

Notifications You must be signed in to change notification settings

vinnik-dmitry07/automl

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

automl

I have trained 5 models in the automl setting.
They are arranged from the worst to the best.
The final decisions were made based on the underlined validation metrics:

keras_automl keyword matrix classifier -- acc: 0.685
keras_automl text classifier -- acc: 0.731
sklearn_automl keyword matrix classifier acc criterion -- acc: 0.741, f1_macro: 0.388, matthew_corr: 0.673
sklearn_automl keyword matrix classifier f1_macro criterion -- acc: 0.724, f1_macro: 0.461, matthew_corr: 0.651
sklearn_automl text classifier f1_macro criterion -- acc: 0.776, f1_macro: 0.457, matthew_corr: 0.722

Also, I trained a perceptron on the keyword matrix:
pytorch lightning keyword matrix classifier -- acc: 0.763, f1_macro: 0.463, matthew_corr: 0.699
and fine-tuned the bert-base-uncased model
pytorch lightning bert text classifier head -- acc: 0.492, f1_macro: 0.134, matthew_corr: 0.311

The perceptron curves:

About

Multiclass classification of doctors from text/keywords

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published