How to use the jupyter.create.ml_template.src.formula.notebook_template.Estimator.KMeans.value function in jupyter

To help you get started, we’ve selected a few jupyter examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

github ZupIT / ritchie-formulas / jupyter / create / ml_template / src / formula / notebook_template.py View on Github external
"evaluation": "regression",
}
content[Estimator.SVR.value] = {
    "import": "from sklearn.svm import SVR",
    "doc_link": "https://scikit-learn.org/\
    stable/modules/generated/sklearn.svm.\
    SVR.html#sklearn.svm.SVR",
    "evaluation": "regression",
}
content[Estimator.MiniBatchKMeans.value] = {
    "import": "from sklearn.cluster import MiniBatchKMeans",
    "doc_link": "https://scikit-learn.org/\
    stable/modules/clustering.html#mini-batch-k-means",
    "evaluation": "classification",
}
content[Estimator.KMeans.value] = {
    "import": "from sklearn.cluster import KMeans",
    "doc_link": "https://scikit-learn.org/\
    stable/modules/clustering.html#k-means",
    "evaluation": "clustering",
}


metrics = {
    "regression": [
        {
            "cell_type": "markdown",
            "metadata": {},
            "source": [
                "## Regression Evaluation Metrics\n",
                "\n",
                "\n",