Esempio n. 1
0
        "type": "array",
        "items": {
            "type": "number"
        }
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.naive_bayes.GaussianNB#sklearn-naive_bayes-gaussiannb",
    "import_from": "sklearn.naive_bayes",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
        "input_predict_proba": _input_predict_proba_schema,
        "output_predict_proba": _output_predict_proba_schema,
    },
}
GaussianNB = make_operator(_GaussianNBImpl, _combined_schemas)

set_docstrings(GaussianNB)
Esempio n. 2
0
    "description": "Transform new points into embedding space.",
    "type": "object",
    "required": ["X"],
    "properties": {
        "X": {"type": "array", "items": {"type": "array", "items": {"type": "number"}}}
    },
}
_output_transform_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Transform new points into embedding space.",
    "type": "array",
    "items": {"type": "array", "items": {"type": "number"}},
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url": "https://scikit-learn.org/0.20/modules/generated/sklearn.manifold.LocallyLinearEmbedding#sklearn-manifold-locallylinearembedding",
    "import_from": "sklearn.manifold",
    "type": "object",
    "tags": {"pre": [], "op": ["transformer"], "post": []},
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_transform": _input_transform_schema,
        "output_transform": _output_transform_schema,
    },
}
LocallyLinearEmbedding = make_operator(_LocallyLinearEmbeddingImpl, _combined_schemas)

set_docstrings(LocallyLinearEmbedding)
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Returns predicted values.",
    "type": "array",
    "items": {
        "type": "number"
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.linear_model.MultiTaskElasticNetCV#sklearn-linear_model-multitaskelasticnetcv",
    "import_from": "sklearn.linear_model",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
MultiTaskElasticNetCV = make_operator(_MultiTaskElasticNetCVImpl,
                                      _combined_schemas)

set_docstrings(MultiTaskElasticNetCV)
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Returns predicted values.",
    "type": "array",
    "items": {
        "type": "number"
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.linear_model.OrthogonalMatchingPursuit#sklearn-linear_model-orthogonalmatchingpursuit",
    "import_from": "sklearn.linear_model",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
OrthogonalMatchingPursuit = make_operator(_OrthogonalMatchingPursuitImpl,
                                          _combined_schemas)

set_docstrings(OrthogonalMatchingPursuit)
Esempio n. 5
0
    "description": "Transform X.",
    "type": "array",
    "items": {
        "type": "array",
        "items": {
            "type": "number"
        }
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.manifold.Isomap#sklearn-manifold-isomap",
    "import_from": "sklearn.manifold",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["transformer"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_transform": _input_transform_schema,
        "output_transform": _output_transform_schema,
    },
}
set_docstrings(IsomapImpl, _combined_schemas)
Isomap = make_operator(IsomapImpl, _combined_schemas)
Esempio n. 6
0
    "description": "Apply approximate feature map to X.",
    "type": "object",
    "required": ["X"],
    "properties": {
        "X": {"type": "array", "items": {"type": "array", "items": {"type": "number"}}}
    },
}
_output_transform_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Whether the return value is an array of sparse matrix depends on the type of the input X.",
    "laleType": "Any",
    "XXX TODO XXX": "{array, sparse matrix},                shape = (n_samples, n_features * (2*sample_steps + 1))",
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url": "https://scikit-learn.org/0.20/modules/generated/sklearn.kernel_approximation.AdditiveChi2Sampler#sklearn-kernel_approximation-additivechi2sampler",
    "import_from": "sklearn.kernel_approximation",
    "type": "object",
    "tags": {"pre": [], "op": ["transformer"], "post": []},
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_transform": _input_transform_schema,
        "output_transform": _output_transform_schema,
    },
}
AdditiveChi2Sampler = make_operator(_AdditiveChi2SamplerImpl, _combined_schemas)

set_docstrings(AdditiveChi2Sampler)
Esempio n. 7
0
    "items": {
        "type": "array",
        "items": {
            "type": "number"
        }
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.neural_network.MLPClassifier#sklearn-neural_network-mlpclassifier",
    "import_from": "sklearn.neural_network",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator", "classifier"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
        "input_predict_proba": _input_predict_proba_schema,
        "output_predict_proba": _output_predict_proba_schema,
    },
}
set_docstrings(MLPClassifierImpl, _combined_schemas)
MLPClassifier = make_operator(MLPClassifierImpl, _combined_schemas)
Esempio n. 8
0
}
_output_predict_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Returns predicted values.",
    "type": "array",
    "items": {
        "type": "number"
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.svm.LinearSVR#sklearn-svm-linearsvr",
    "import_from": "sklearn.svm",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
set_docstrings(LinearSVRImpl, _combined_schemas)
LinearSVR = make_operator(LinearSVRImpl, _combined_schemas)
Esempio n. 9
0
        "type": "array",
        "items": {
            "type": "number"
        }
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.semi_supervised.LabelPropagation#sklearn-semi_supervised-labelpropagation",
    "import_from": "sklearn.semi_supervised",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator", "classifier"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
        "input_predict_proba": _input_predict_proba_schema,
        "output_predict_proba": _output_predict_proba_schema,
    },
}
LabelPropagation = make_operator(_LabelPropagationImpl, _combined_schemas)

set_docstrings(LabelPropagation)
Esempio n. 10
0
        "type": "array",
        "items": {
            "type": "number"
        }
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.gaussian_process.GaussianProcessClassifier#sklearn-gaussian_process-gaussianprocessclassifier",
    "import_from": "sklearn.gaussian_process",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator", "classifier"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
        "input_predict_proba": _input_predict_proba_schema,
        "output_predict_proba": _output_predict_proba_schema,
    },
}
set_docstrings(GaussianProcessClassifierImpl, _combined_schemas)
GaussianProcessClassifier = make_operator(GaussianProcessClassifierImpl,
                                          _combined_schemas)
Esempio n. 11
0
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Index of the cluster each sample belongs to.",
    "type": "array",
    "items": {
        "type": "number"
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.cluster.MiniBatchKMeans#sklearn-cluster-minibatchkmeans",
    "import_from": "sklearn.cluster",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["transformer", "estimator"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_transform": _input_transform_schema,
        "output_transform": _output_transform_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
set_docstrings(MiniBatchKMeansImpl, _combined_schemas)
MiniBatchKMeans = make_operator(MiniBatchKMeansImpl, _combined_schemas)
Esempio n. 12
0
}
_output_transform_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "The missing indicator for input data",
    "laleType": "Any",
    "XXX TODO XXX":
    "{ndarray or sparse matrix}, shape (n_samples, n_features)",
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.impute.MissingIndicator#sklearn-impute-missingindicator",
    "import_from": "sklearn.impute",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["transformer"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_transform": _input_transform_schema,
        "output_transform": _output_transform_schema,
    },
}
MissingIndicator = make_operator(_MissingIndicatorImpl, _combined_schemas)

set_docstrings(MissingIndicator)
Esempio n. 13
0
        },
    },
}
_output_predict_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Predict using the linear model.",
    "laleType": "Any",
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.linear_model.ARDRegression#sklearn-linear_model-ardregression",
    "import_from": "sklearn.linear_model",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
ARDRegression = make_operator(_ARDRegressionImpl, _combined_schemas)

set_docstrings(ARDRegression)
Esempio n. 14
0
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Index of the cluster each sample belongs to.",
    "type": "array",
    "items": {
        "type": "number"
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.cluster.KMeans#sklearn-cluster-kmeans",
    "import_from": "sklearn.cluster",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["transformer", "estimator"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_transform": _input_transform_schema,
        "output_transform": _output_transform_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
set_docstrings(KMeansImpl, _combined_schemas)
KMeans = make_operator(KMeansImpl, _combined_schemas)
Esempio n. 15
0
                    "items": {"type": "array", "items": {"type": "number"}},
                },
            ],
            "description": "Samples.",
        }
    },
}
_output_predict_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Returns predicted values.",
    "type": "array",
    "items": {"type": "number"},
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url": "https://scikit-learn.org/0.20/modules/generated/sklearn.linear_model.LassoCV#sklearn-linear_model-lassocv",
    "import_from": "sklearn.linear_model",
    "type": "object",
    "tags": {"pre": [], "op": ["estimator", "regressor"], "post": []},
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
LassoCV = make_operator(_LassoCVImpl, _combined_schemas)

set_docstrings(LassoCV)
Esempio n. 16
0
    "items": {
        "type": "array",
        "items": {
            "type": "number"
        }
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.semi_supervised.LabelSpreading#sklearn-semi_supervised-labelspreading",
    "import_from": "sklearn.semi_supervised",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
        "input_predict_proba": _input_predict_proba_schema,
        "output_predict_proba": _output_predict_proba_schema,
    },
}
set_docstrings(LabelSpreadingImpl, _combined_schemas)
LabelSpreading = make_operator(LabelSpreadingImpl, _combined_schemas)
Esempio n. 17
0
            "description": "The data used to scale along the specified axis.",
        }
    },
}
_output_transform_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Center and scale the data.",
    "laleType": "Any",
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.preprocessing.RobustScaler#sklearn-preprocessing-robustscaler",
    "import_from": "sklearn.preprocessing",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["transformer"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_transform": _input_transform_schema,
        "output_transform": _output_transform_schema,
    },
}
set_docstrings(RobustScalerImpl, _combined_schemas)
RobustScaler = make_operator(RobustScalerImpl, _combined_schemas)
Esempio n. 18
0
            ],
            "description": "Samples.",
        }
    },
}
_output_decision_function_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Confidence scores per (sample, class) combination",
    "laleType": "Any",
    "XXX TODO XXX": "array, shape=(n_samples,) if n_classes == 2 else (n_samples, n_classes)",
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url": "https://scikit-learn.org/0.20/modules/generated/sklearn.linear_model.RidgeClassifierCV#sklearn-linear_model-ridgeclassifiercv",
    "import_from": "sklearn.linear_model",
    "type": "object",
    "tags": {"pre": [], "op": ["estimator"], "post": []},
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
        "input_decision_function": _input_decision_function_schema,
        "output_decision_function": _output_decision_function_schema,
    },
}
RidgeClassifierCV = make_operator(_RidgeClassifierCVImpl, _combined_schemas)

set_docstrings(RidgeClassifierCV)
Esempio n. 19
0
        "type": "array",
        "items": {
            "type": "number"
        }
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.naive_bayes.ComplementNB#sklearn-naive_bayes-complementnb",
    "import_from": "sklearn.naive_bayes",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
        "input_predict_proba": _input_predict_proba_schema,
        "output_predict_proba": _output_predict_proba_schema,
    },
}
ComplementNB = make_operator(_ComplementNBImpl, _combined_schemas)

set_docstrings(ComplementNB)
Esempio n. 20
0
File: nu_svc.py Progetto: lnxpy/lale
    "Returns the decision function of the sample for each class in the model",
    "laleType": "Any",
    "XXX TODO XXX":
    "array-like, shape (n_samples, n_classes * (n_classes-1) / 2)",
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.svm.NuSVC#sklearn-svm-nusvc",
    "import_from": "sklearn.svm",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
        "input_predict_proba": _input_predict_proba_schema,
        "output_predict_proba": _output_predict_proba_schema,
        "input_decision_function": _input_decision_function_schema,
        "output_decision_function": _output_decision_function_schema,
    },
}
set_docstrings(NuSVCImpl, _combined_schemas)
NuSVC = make_operator(NuSVCImpl, _combined_schemas)
Esempio n. 21
0
    "description": "Latent representations of the data.",
    "type": "array",
    "items": {
        "type": "array",
        "items": {
            "type": "number"
        }
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.neural_network.BernoulliRBM#sklearn-neural_network-bernoullirbm",
    "import_from": "sklearn.neural_network",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["transformer"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_transform": _input_transform_schema,
        "output_transform": _output_transform_schema,
    },
}
set_docstrings(BernoulliRBMImpl, _combined_schemas)
BernoulliRBM = make_operator(BernoulliRBMImpl, _combined_schemas)
Esempio n. 22
0
            "items": {"type": "array", "items": {"type": "number"}},
            "description": "Samples",
        }
    },
}
_output_predict_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Returns predicted values.",
    "anyOf": [
        {"type": "array", "items": {"type": "number"}},
        {"type": "array", "items": {"type": "array", "items": {"type": "number"}}},
    ],
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url": "https://scikit-learn.org/0.20/modules/generated/sklearn.kernel_ridge.KernelRidge#sklearn-kernel_ridge-kernelridge",
    "import_from": "sklearn.kernel_ridge",
    "type": "object",
    "tags": {"pre": [], "op": ["estimator"], "post": []},
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
KernelRidge = make_operator(_KernelRidgeImpl, _combined_schemas)

set_docstrings(KernelRidge)
Esempio n. 23
0
                    "items": {"type": "array", "items": {"type": "number"}},
                },
            ],
            "description": "Samples.",
        }
    },
}
_output_predict_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Returns predicted values.",
    "type": "array",
    "items": {"type": "number"},
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url": "https://scikit-learn.org/0.20/modules/generated/sklearn.linear_model.LinearRegression#sklearn-linear_model-linearregression",
    "import_from": "sklearn.linear_model",
    "type": "object",
    "tags": {"pre": [], "op": ["estimator"], "post": []},
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
LinearRegression = make_operator(_LinearRegressionImpl, _combined_schemas)

set_docstrings(LinearRegression)
Esempio n. 24
0
    "Confidence scores per (sample, class) combination",
    "laleType":
    "Any",
    "XXX TODO XXX":
    "array, shape=(n_samples,) if n_classes == 2 else (n_samples, n_classes)",
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.svm.LinearSVC#sklearn-svm-linearsvc",
    "import_from": "sklearn.svm",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
        "input_decision_function": _input_decision_function_schema,
        "output_decision_function": _output_decision_function_schema,
    },
}
LinearSVC = make_operator(_LinearSVCImpl, _combined_schemas)

set_docstrings(LinearSVC)
Esempio n. 25
0
    "type": "array",
    "items": {
        "type": "array",
        "items": {
            "type": "number"
        }
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.decomposition.DictionaryLearning#sklearn-decomposition-dictionarylearning",
    "import_from": "sklearn.decomposition",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["transformer"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_transform": _input_transform_schema,
        "output_transform": _output_transform_schema,
    },
}
DictionaryLearning = make_operator(_DictionaryLearningImpl, _combined_schemas)

set_docstrings(DictionaryLearning)
Esempio n. 26
0
}
_output_predict_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Returns predicted values.",
    "type": "array",
    "items": {
        "type": "number"
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.linear_model.LinearRegression#sklearn-linear_model-linearregression",
    "import_from": "sklearn.linear_model",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
set_docstrings(LinearRegressionImpl, _combined_schemas)
LinearRegression = make_operator(LinearRegressionImpl, _combined_schemas)
Esempio n. 27
0
_output_predict_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Predicted values.",
    "type": "array",
    "items": {
        "type": "number"
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.compose.TransformedTargetRegressor#sklearn-compose-transformedtargetregressor",
    "import_from": "sklearn.compose",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator", "regressor"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
set_docstrings(TransformedTargetRegressorImpl, _combined_schemas)
TransformedTargetRegressor = make_operator(TransformedTargetRegressorImpl,
                                           _combined_schemas)
Esempio n. 28
0
        "copy": {
            "type": "boolean",
            "default": True,
            "description": "Whether to copy X and Y, or perform in-place normalization.",
        },
    },
}
_output_predict_schema = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Apply the dimension reduction learned on the train data.",
    "laleType": "Any",
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url": "https://scikit-learn.org/0.20/modules/generated/sklearn.cross_decomposition.PLSRegression#sklearn-cross_decomposition-plsregression",
    "import_from": "sklearn.cross_decomposition",
    "type": "object",
    "tags": {"pre": [], "op": ["transformer", "estimator"], "post": []},
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_transform": _input_transform_schema,
        "output_transform": _output_transform_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
set_docstrings(PLSRegressionImpl, _combined_schemas)
PLSRegression = make_operator(PLSRegressionImpl, _combined_schemas)
Esempio n. 29
0
            "items": {
                "type": "array",
                "items": {
                    "type": "number"
                }
            }
        },
    ],
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.ensemble.ExtraTreesRegressor#sklearn-ensemble-extratreesregressor",
    "import_from": "sklearn.ensemble",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["estimator", "regressor"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_predict": _input_predict_schema,
        "output_predict": _output_predict_schema,
    },
}
set_docstrings(ExtraTreesRegressorImpl, _combined_schemas)
ExtraTreesRegressor = make_operator(ExtraTreesRegressorImpl, _combined_schemas)
Esempio n. 30
0
    "type": "array",
    "items": {
        "type": "array",
        "items": {
            "type": "number"
        }
    },
}
_combined_schemas = {
    "$schema": "http://json-schema.org/draft-04/schema#",
    "description": "Combined schema for expected data and hyperparameters.",
    "documentation_url":
    "https://scikit-learn.org/0.20/modules/generated/sklearn.preprocessing.PowerTransformer#sklearn-preprocessing-powertransformer",
    "import_from": "sklearn.preprocessing",
    "type": "object",
    "tags": {
        "pre": [],
        "op": ["transformer"],
        "post": []
    },
    "properties": {
        "hyperparams": _hyperparams_schema,
        "input_fit": _input_fit_schema,
        "input_transform": _input_transform_schema,
        "output_transform": _output_transform_schema,
    },
}
PowerTransformer = make_operator(_PowerTransformerImpl, _combined_schemas)

set_docstrings(PowerTransformer)