search

ClimbsRocks / auto_ml

[UNMAINTAINED] Automated machine learning for analytics & production
http://auto-ml.readthedocs.io
MIT License
1.64k stars 310 forks source link

Multiclass with DeepLearningClassifier #430

Open davidserranogemes opened 5 years ago

davidserranogemes commented 5 years ago

Hi, I have been trying to use this project for a multiclass problems. When using the DeepLearningClass, the acc is near constant at 20%. Looking at the predictions, near all values are 1 when there are 10 classes.

When using the default model it works fine.

I have just read this https://github.com/ClimbsRocks/auto_ml/issues/76

I would like to get confirmation about if DeepLearningClassifier suppors multiclassification or not.

Thanks in advance!