nrel-shift 0.1.0a0__tar.gz

NREL-shift-0.1.0a0/LICENSE.txt

@@ -0,0 +1,29 @@
+BSD 3-Clause License
+
+Copyright (c) 2022, Alliance for Sustainable Energy, LLC
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its
+   contributors may be used to endorse or promote products derived from
+   this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

NREL-shift-0.1.0a0/NREL_shift.egg-info/PKG-INFO

@@ -0,0 +1,45 @@
+Metadata-Version: 2.1
+Name: NREL-shift
+Version: 0.1.0a0
+Summary: Generate synthetic feeders using open street map data
+Home-page: https://github.com/nrel/shift
+Author: Kapil Duwadi
+Author-email: kapil.duwadi@nrel.gov
+Keywords: Synthetic feeder,Open steet data,OpenDSS
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Programming Language :: Python :: 3.7
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE.txt
+
+
+Simple Synthetic Distribution Feeder Generation Tool (SHIFT)
+
+<img src="docs/images/shift.svg" width="400" style="display:block;margin:auto;">
+
+
+![GitHub all releases](https://img.shields.io/github/downloads/NREL/shift/total?logo=Github&logoColor=%2300ff00&style=flat-square)
+![GitHub repo size](https://img.shields.io/github/repo-size/nrel/shift?style=flat-square)
+![CodeFactor Grade](https://img.shields.io/codefactor/grade/github/nrel/shift?color=%23ff0000&logo=python&logoColor=%2300ff00&style=flat-square)
+[![GitHub license](https://img.shields.io/github/license/NREL/shift?style=flat-square)](https://github.com/NREL/shift/blob/main/LICENSE.txt)
+[![GitHub issues](https://img.shields.io/github/issues/NREL/shift?style=flat-square)](https://github.com/NREL/shift/issues)
+![GitHub top language](https://img.shields.io/github/languages/top/nrel/shift?style=flat-square)
+![Snyk Vulnerabilities for GitHub Repo](https://img.shields.io/snyk/vulnerabilities/github/nrel/shift?style=flat-square)
+
+### :wave:  Welcome to SHIFT repository!
+
+[Goto full documentation](https://nrel.github.io/shift/)
+
+Getting free distribution feeder models for your reasearch has never been easier.
+SHIFT helps you build synthetic distribution feeders models using just the OpenStreet data e.g. buildings and road network. You can configure lots of
+parameters and design choices when building these models and even create multiple versions of them and choose the one that fits your needs. If you are utility no worries we 
+have you covered as well. As a utility you can also integrate your data when building these models. 
+
+Having distribution models in your hand can help you perform rich analysis in understanding impact of distributed energy resources 
+such as roof top PV, energy storage (Tesla Wall, lead acid, lithium name ....), electric vehicle and more. Performing network upgrade analysis, hosting capacity 
+analysis, interconnection studies all require detailed distribution models.
+
+So what are you waiting for. We can't wait to see how you will use this tool for your as wells your organizations's benefit.
+
+Feel free to reach out to us or send us a comment using Github issue. We don't mind if you smash that starred icon motivating our developers and contributors.

NREL-shift-0.1.0a0/NREL_shift.egg-info/SOURCES.txt

@@ -0,0 +1,35 @@
+LICENSE.txt
+README.md
+setup.py
+NREL_shift.egg-info/PKG-INFO
+NREL_shift.egg-info/SOURCES.txt
+NREL_shift.egg-info/dependency_links.txt
+NREL_shift.egg-info/entry_points.txt
+NREL_shift.egg-info/requires.txt
+NREL_shift.egg-info/top_level.txt
+shift/__init__.py
+shift/clustering.py
+shift/config_template.py
+shift/constants.py
+shift/enums.py
+shift/exceptions.py
+shift/facade.py
+shift/feeder_network.py
+shift/geometry.py
+shift/graph.py
+shift/line_section.py
+shift/load.py
+shift/load_builder.py
+shift/network_plots.py
+shift/primary_network_builder.py
+shift/secondary_network_builder.py
+shift/transformer.py
+shift/transformer_builder.py
+shift/utils.py
+shift/cli/__init__.py
+shift/cli/shift.py
+shift/examples/__init__.py
+shift/examples/example0.py
+shift/exporter/__init__.py
+shift/exporter/base.py
+shift/exporter/opendss.py

NREL-shift-0.1.0a0/NREL_shift.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

NREL-shift-0.1.0a0/NREL_shift.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+shift-internal = shift.cli.shift:cli

NREL-shift-0.1.0a0/NREL_shift.egg-info/requires.txt

@@ -0,0 +1,18 @@
+osmnx
+cerberus
+geopy
+pyshp
+shapely
+plotly
+numpy
+networkx
+pandas
+sklearn
+pydantic
+click
+pytest
+openpyxl
+mkdocs
+mkdocstrings[python]
+mkdocs-material
+mkdocs-jupyter

NREL-shift-0.1.0a0/NREL_shift.egg-info/top_level.txt

@@ -0,0 +1 @@
+shift

NREL-shift-0.1.0a0/PKG-INFO

@@ -0,0 +1,45 @@
+Metadata-Version: 2.1
+Name: NREL-shift
+Version: 0.1.0a0
+Summary: Generate synthetic feeders using open street map data
+Home-page: https://github.com/nrel/shift
+Author: Kapil Duwadi
+Author-email: kapil.duwadi@nrel.gov
+Keywords: Synthetic feeder,Open steet data,OpenDSS
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Programming Language :: Python :: 3.7
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE.txt
+
+
+Simple Synthetic Distribution Feeder Generation Tool (SHIFT)
+
+<img src="docs/images/shift.svg" width="400" style="display:block;margin:auto;">
+
+
+![GitHub all releases](https://img.shields.io/github/downloads/NREL/shift/total?logo=Github&logoColor=%2300ff00&style=flat-square)
+![GitHub repo size](https://img.shields.io/github/repo-size/nrel/shift?style=flat-square)
+![CodeFactor Grade](https://img.shields.io/codefactor/grade/github/nrel/shift?color=%23ff0000&logo=python&logoColor=%2300ff00&style=flat-square)
+[![GitHub license](https://img.shields.io/github/license/NREL/shift?style=flat-square)](https://github.com/NREL/shift/blob/main/LICENSE.txt)
+[![GitHub issues](https://img.shields.io/github/issues/NREL/shift?style=flat-square)](https://github.com/NREL/shift/issues)
+![GitHub top language](https://img.shields.io/github/languages/top/nrel/shift?style=flat-square)
+![Snyk Vulnerabilities for GitHub Repo](https://img.shields.io/snyk/vulnerabilities/github/nrel/shift?style=flat-square)
+
+### :wave:  Welcome to SHIFT repository!
+
+[Goto full documentation](https://nrel.github.io/shift/)
+
+Getting free distribution feeder models for your reasearch has never been easier.
+SHIFT helps you build synthetic distribution feeders models using just the OpenStreet data e.g. buildings and road network. You can configure lots of
+parameters and design choices when building these models and even create multiple versions of them and choose the one that fits your needs. If you are utility no worries we 
+have you covered as well. As a utility you can also integrate your data when building these models. 
+
+Having distribution models in your hand can help you perform rich analysis in understanding impact of distributed energy resources 
+such as roof top PV, energy storage (Tesla Wall, lead acid, lithium name ....), electric vehicle and more. Performing network upgrade analysis, hosting capacity 
+analysis, interconnection studies all require detailed distribution models.
+
+So what are you waiting for. We can't wait to see how you will use this tool for your as wells your organizations's benefit.
+
+Feel free to reach out to us or send us a comment using Github issue. We don't mind if you smash that starred icon motivating our developers and contributors.

NREL-shift-0.1.0a0/README.md

@@ -0,0 +1,30 @@
+
+Simple Synthetic Distribution Feeder Generation Tool (SHIFT)
+
+<img src="docs/images/shift.svg" width="400" style="display:block;margin:auto;">
+
+
+![GitHub all releases](https://img.shields.io/github/downloads/NREL/shift/total?logo=Github&logoColor=%2300ff00&style=flat-square)
+![GitHub repo size](https://img.shields.io/github/repo-size/nrel/shift?style=flat-square)
+![CodeFactor Grade](https://img.shields.io/codefactor/grade/github/nrel/shift?color=%23ff0000&logo=python&logoColor=%2300ff00&style=flat-square)
+[![GitHub license](https://img.shields.io/github/license/NREL/shift?style=flat-square)](https://github.com/NREL/shift/blob/main/LICENSE.txt)
+[![GitHub issues](https://img.shields.io/github/issues/NREL/shift?style=flat-square)](https://github.com/NREL/shift/issues)
+![GitHub top language](https://img.shields.io/github/languages/top/nrel/shift?style=flat-square)
+![Snyk Vulnerabilities for GitHub Repo](https://img.shields.io/snyk/vulnerabilities/github/nrel/shift?style=flat-square)
+
+### :wave:  Welcome to SHIFT repository!
+
+[Goto full documentation](https://nrel.github.io/shift/)
+
+Getting free distribution feeder models for your reasearch has never been easier.
+SHIFT helps you build synthetic distribution feeders models using just the OpenStreet data e.g. buildings and road network. You can configure lots of
+parameters and design choices when building these models and even create multiple versions of them and choose the one that fits your needs. If you are utility no worries we 
+have you covered as well. As a utility you can also integrate your data when building these models. 
+
+Having distribution models in your hand can help you perform rich analysis in understanding impact of distributed energy resources 
+such as roof top PV, energy storage (Tesla Wall, lead acid, lithium name ....), electric vehicle and more. Performing network upgrade analysis, hosting capacity 
+analysis, interconnection studies all require detailed distribution models.
+
+So what are you waiting for. We can't wait to see how you will use this tool for your as wells your organizations's benefit.
+
+Feel free to reach out to us or send us a comment using Github issue. We don't mind if you smash that starred icon motivating our developers and contributors.

NREL-shift-0.1.0a0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

NREL-shift-0.1.0a0/setup.py

@@ -0,0 +1,30 @@
+import os
+from setuptools import setup, find_packages
+
+with open("README.md", "r") as fh:
+    long_description = fh.read()
+
+with open(os.path.join(os.path.dirname(__file__), "requirements.txt")) as f:
+    requirements = f.read().splitlines()
+
+setup(
+    name="NREL-shift",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    version="v0.1.0-alpha",
+    description="Generate synthetic feeders using open street map data",
+    author="Kapil Duwadi",
+    author_email="kapil.duwadi@nrel.gov",
+    packages=find_packages(),
+    url="https://github.com/nrel/shift",
+    keywords="Synthetic feeder, Open steet data, OpenDSS",
+    install_requires=requirements,
+    python_requires=">=3.9",
+    package_dir={"shift": "shift"},
+    entry_points={"console_scripts": ["shift-internal=shift.cli.shift:cli"]},
+    classifiers=[
+        "License :: OSI Approved :: BSD License",
+        "Programming Language :: Python :: 3.7",
+        "Operating System :: OS Independent",
+    ],
+)

NREL-shift-0.1.0a0/shift/cli/shift.py

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+# Copyright (c) 2022, Alliance for Sustainable Energy, LLC
+
+# All rights reserved.
+
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+
+# 1. Redistributions of source code must retain the above copyright notice, this
+#    list of conditions and the following disclaimer.
+
+# 2. Redistributions in binary form must reproduce the above copyright notice,
+#    this list of conditions and the following disclaimer in the documentation
+#    and/or other materials provided with the distribution.
+
+# 3. Neither the name of the copyright holder nor the names of its
+#    contributors may be used to endorse or promote products derived from
+#    this software without specific prior written permission.
+
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+"""This module handles command line interface for this package. """
+import json
+
+import click
+import yaml
+
+from shift.config_template import ConfigTemplate
+from shift.facade import generate_feeder_from_yaml
+
+
+@click.command()
+@click.option("-c", "--config-yaml", help="Path to config yaml file.")
+def create_config_file(config_yaml: str) -> None:
+    """Creates a default config yaml file. Update this yaml file and use it to
+    create the synthetic distribution feeder.
+    """
+    config = ConfigTemplate().dict()
+    print(json.loads(json.dumps(config)))
+    with open(config_yaml, "w", encoding="utf-8") as fpointer:
+        yaml.dump(json.loads(json.dumps(config)), fpointer)
+
+
+@click.command()
+@click.option("-c", "--config-yaml", help="Path to config yaml file.")
+def create_feeder(config_yaml: str) -> None:
+    """Creates a synthetic distribution feeder
+    by taking the config yaml file as an input.
+    """
+    generate_feeder_from_yaml(config_yaml)
+
+
+@click.group()
+def cli():
+    """Entry point"""
+    pass
+
+
+cli.add_command(create_config_file)
+cli.add_command(create_feeder)

NREL-shift-0.1.0a0/shift/clustering.py

@@ -0,0 +1,229 @@
+# -*- coding: utf-8 -*-
+# Copyright (c) 2022, Alliance for Sustainable Energy, LLC
+
+# All rights reserved.
+
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+
+# 1. Redistributions of source code must retain the above copyright notice, this
+#    list of conditions and the following disclaimer.
+
+# 2. Redistributions in binary form must reproduce the above copyright notice,
+#    this list of conditions and the following disclaimer in the documentation
+#    and/or other materials provided with the distribution.
+
+# 3. Neither the name of the copyright holder nor the names of its
+#    contributors may be used to endorse or promote products derived from
+#    this software without specific prior written permission.
+
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+""" Contains abstract and concrete classes for implementing clustering.
+
+This module contains classes for managing clustering of location pairs.
+The cluster center is used as reference location to figure out
+transformer siting. In the process mapping between
+cluster center and locations is also created.
+
+Examples:
+    Get the clusters from given list of location pairs and plot it.
+
+    >>> from shift.clustering import KmeansClustering
+    >>> import numpy as np
+    >>> x_array = np.array([[1,2], [3,4], [5,6]])
+    >>> cluster = KmeansClustering(2)
+    >>> cluster.get_clusters(x_array)
+    >>> cluster.plot_clusters()
+"""
+
+from abc import ABC, abstractmethod
+from typing import List, Union
+
+from sklearn.cluster import KMeans
+from sklearn.metrics import silhouette_score
+import plotly.express as px
+import pandas as pd
+
+from shift.constants import MAX_KMEANS_LOOP, MIN_NUM_CLUSTER
+from shift.exceptions import (
+    MaxLoopReachedForKmeans,
+    WrongInputUsed,
+    NumberOfClusterNotInRangeError,
+    EarlyMethodCallError,
+)
+
+
+class Clustering(ABC):
+    """Abstract interface for implementing clustering subclass"""
+
+    @abstractmethod
+    def get_clusters(self, x_array: List[list]) -> dict:
+        """Method for creating and returning clusters.
+
+        Subclass inhereting this class must implement this method and should
+        return the clusters in the following format.
+        ```json
+        {"labels": [0,1,1], "centre": [(0,0), (2,2)]}
+        ```
+
+        Returns:
+            dict: A dictionary containing cluster labels
+                and cluster centers
+        """
+        pass
+
+    @abstractmethod
+    def plot_clusters(self) -> None:
+        """Method to plot clusters
+
+        Raises:
+            EarlyMethodCallError: If this method is called before calling
+                `get_clusters` method.
+        """
+        pass
+
+
+class KmeansClustering(Clustering):
+    """Class implementing Kmeans clustering.
+
+    Attributes:
+        num_of_clusters (Union[str, int]): Number of clusters to be used
+        cluster_centers (List[Sequece]): List of cluster center
+        labels (list): Integer label for each location indicating which
+            cluster they belong to
+        xarray (List[Sequece]): List of location pairs for which
+            clustering is performed
+        optimal_clusters (int): Optimal number of clusters created if
+            `num_of_clusters` passed is `optimal`
+    """
+
+    def __init__(self, num_of_clusters: Union[str, int] = "optimal") -> None:
+        """Constructor for `KmeansClustering` class.
+
+        Args:
+            num_of_clusters (Union[str, int]): Number of clusters to be used
+
+        Raises:
+            NumberOfClusterNotInRangeError: if `num_of_clusters` speecified is
+                less than MIN_NUM_CLUSTER constants module.
+        """
+        self.num_of_clusters = num_of_clusters
+
+        if isinstance(self.num_of_clusters, int):
+            if self.num_of_clusters < MIN_NUM_CLUSTER:
+                raise NumberOfClusterNotInRangeError(self.num_of_clusters)
+
+    def plot_clusters(self):
+        """Refer to the base class for details."""
+
+        if (
+            hasattr(self, "x_array")
+            and hasattr(self, "labels")
+            and hasattr(self, "cluster_centers")
+        ):
+            plot_data = {"x": [], "y": [], "label": []}
+            for d, label in zip(self.x_array, self.labels):
+                plot_data["x"].append(d[0])
+                plot_data["y"].append(d[1])
+                plot_data["label"].append(label)
+            fig = px.scatter(
+                pd.DataFrame(plot_data), x="x", y="y", color="label"
+            )
+            fig.show()
+
+        else:
+            raise EarlyMethodCallError(
+                "Call get_clusters() method first before "
+                + "calling plot clusters method!"
+            )
+
+    def plot_scores(self):
+        """Method for plotting scores
+
+        Note:
+            Only use this method if `num_of_clusters` passed is `optimal`.
+
+        Raises:
+            EarlyMethodCallError: If called before calling
+                `get_clusters` method.
+        """
+
+        if hasattr(self, "sil_scores"):
+            plot_data = {
+                "Number of clusters": [x[0] for x in self.sil_scores],
+                "Silhouette Value": [x[1] for x in self.sil_scores],
+            }
+            fig = px.line(
+                pd.DataFrame(plot_data),
+                x="Number of clusters",
+                y="Silhouette Value",
+            )
+            fig.show()
+
+        else:
+            raise EarlyMethodCallError(
+                "Call get_clusters() method first before calling "
+                + "plot scores method!"
+            )
+
+    def get_clusters(self, x_array: list) -> dict:
+        """Refer to the base class for details.
+
+        Raises:
+            WrongInputUsed: If the number of clusters passed is invalid.
+        """
+
+        self.x_array = x_array
+        if isinstance(self.num_of_clusters, int):
+            kmeans = KMeans(
+                n_clusters=self.num_of_clusters, random_state=0
+            ).fit(x_array)
+
+        else:
+            if self.num_of_clusters == "optimal":
+                # Let's try to find optimal number of clusters
+                self.sil_scores = []
+
+                for k in range(2, MAX_KMEANS_LOOP):
+                    kmeans = KMeans(n_clusters=k).fit(x_array)
+                    labels = kmeans.labels_
+                    self.sil_scores.append(
+                        (
+                            k,
+                            silhouette_score(
+                                x_array, labels, metric="euclidean"
+                            ),
+                        )
+                    )
+
+                    # Break the loop if Silhouette score starts to decrease
+                    if k > 2:
+                        if (
+                            self.sil_scores[k - 2][1]
+                            < self.sil_scores[k - 3][1]
+                        ):
+                            break
+
+                if k == MAX_KMEANS_LOOP - 1:
+                    raise MaxLoopReachedForKmeans()
+                kmeans = KMeans(n_clusters=k - 1, random_state=0).fit(x_array)
+                self.optimal_clusters = k - 1
+            else:
+                raise WrongInputUsed(
+                    "For now  number of clusters can be either integer "
+                    + "number or 'optimal'!"
+                    + f"You provided {self.num_of_clusters}"
+                )
+        self.labels = kmeans.labels_
+        self.cluster_centers = kmeans.cluster_centers_
+        return {"labels": self.labels, "centre": self.cluster_centers}