gemseo / mlearning / classification

factory module

Classification model factory

This module contains a factory to instantiate a MLClassificationAlgo from its class name. The class can be internal to GEMSEO or located in an external module whose path is provided to the constructor. It also provides a list of available classification models and allows you to test if a classification model type is available.

class gemseo.mlearning.classification.factory.ClassificationModelFactory[source]

Bases: gemseo.mlearning.core.factory.MLAlgoFactory

This factory instantiates a MLRegressionAlgo from its class name. The class can be internal to GEMSEO or located in an external module whose path is provided to the constructor.

Initializes the factory: scans the directories to search for subclasses of MLClassificationAlgo. Searches in “GEMSEO_PATH” and gemseo.mlearning.classification.