search

ahmedfgad / GeneticAlgorithmPython

Source code of PyGAD, a Python 3 library for building the genetic algorithm and training machine learning algorithms (Keras & PyTorch).
https://pygad.readthedocs.io
BSD 3-Clause "New" or "Revised" License
1.83k stars 457 forks source link

Multi-label output support? #113

Open Alex-Bujorianu opened 2 years ago

Alex-Bujorianu commented 2 years ago

Does PyGAD support multi-label classification? That is, classification where the classes are not mutually exclusive? My y variable is a numpy array created by the MultiLabelBinarizer. I am trying to build a recommendation algorithm for patients, and the patients typically receive anywhere from 1 to 5 recommendations.

ahmedfgad commented 1 year ago

PyGAD supports training Keras and PyTorch models regardless of the problem. So, yes it supports multi-class problems.