search

salesforce / TransmogrifAI

TransmogrifAI (pronounced trăns-mŏgˈrə-fī) is an AutoML library for building modular, reusable, strongly typed machine learning workflows on Apache Spark with minimal hand-tuning
https://transmogrif.ai
BSD 3-Clause "New" or "Revised" License
2.24k stars 392 forks source link

MultiClassClassificationModelsToTry and BinaryClassificationModelsToTry not contains OpMultilayerPerceptronClassifier #518

Open hjfrank1991 opened 3 years ago

tovbinm commented 3 years ago

MLP was not evaluated during our experiments. In order to add it to one of our model selectors, we would need to define a parameter grid. Would you recomend a good set of parameters for it?

Otherwise, you can still use MLP directly without the model selector.

hjfrank1991 commented 3 years ago

when i use MultiClassificationModelSelector i want use OpGBTClassifier Or other custom models, what should I do extends BinaryClassificationModelsToTry but it is sealed trait