tpot2
tpot2 copied to clipboard
allow individual modules to be specified by string for config dicts. For example ["LogisticRegression", "KNN"] etc.
Add a function that takes in a string name for a sklearn pipeline or the sklearn type (e.g "LogisticRegression" OR sklearn.linear_model.LogisticRegression) and returns the appropriate params function. So that we can support users easily choosing a specific set of operators one by one. (e.g. root_config_dict=["logisticregression", "randomforest"]). This function should take the following parameters name, n_samples, n_features, classification, subsets=None, feature_names=None, n_classes=None. It will be used in this function https://github.com/EpistasisLab/tpot2/blob/main/tpot2/tpot_estimator/estimator_utils.py#L24C43-L24C130