Source code for lcc.stars_processing.deciders.unsupervised.k_means_decider

from matplotlib import pyplot as plt
from sklearn.cluster.k_means_ import KMeans

import numpy as np
from lcc.stars_processing.utilities.unsupervised_base import UnsupervisedBase


[docs]class KMeansDecider(UnsupervisedBase): ''' classdocs ''' def __init__(self, treshold=0.5, n_clusters=3): ''' Constructor ''' params = {"n_clusters": n_clusters} super(KMeansDecider, self).__init__( classifier=KMeans, params=params, treshold=treshold)