Source code for lcc.stars_processing.utilities.unsupervised_base

from matplotlib import pyplot as plt

import numpy as np
from lcc.stars_processing.utilities.base_decider import BaseDecider
from lcc.entities.exceptions import QueryInputError


[docs]class UnsupervisedBase(BaseDecider): ''' classdocs ''' def __init__(self, classifier, params, treshold=0.5, **kwargs): super(UnsupervisedBase, self).__init__(**kwargs) self.classifier = classifier(**params)
[docs] def learn(self, coords): coords = [c for c in coords if not np.NaN in c and not None in c] if coords: self.X = np.array(coords) self.classifier.fit(coords) else: raise QueryInputError("No coordinates for learning")
[docs] def evaluate(self, star_coords): return self.classifier.predict(star_coords)