mirror of
https://github.com/prise6/smart-iss-posts
synced 2024-05-05 07:03:10 +02:00
35 lines
1,002 B
Python
35 lines
1,002 B
Python
# -*- coding: utf-8 -*-
|
|
from iss.tools import Tools
|
|
|
|
class AbstractClustering:
|
|
|
|
def __init__(self, config, pictures_id, pictures_np):
|
|
|
|
self.config = config
|
|
self.pictures_id = pictures_id
|
|
self.pictures_np = pictures_np
|
|
self.final_labels = None
|
|
self.colors = None
|
|
|
|
if self.config['save_directory']:
|
|
Tools.create_dir_if_not_exists(self.config['save_directory'])
|
|
|
|
def compute_final_labels(self):
|
|
raise NotImplementedError
|
|
|
|
def get_results(self):
|
|
return list(zip(self.pictures_id, self.final_labels, self.pictures_np))
|
|
|
|
def compute_silhouette_score(self):
|
|
raise NotImplementedError
|
|
|
|
def compute_colors(self):
|
|
n_classes = len(list(set(self.final_labels)))
|
|
self.colors = [Tools.get_color_from_label(label, n_classes) for label in self.final_labels]
|
|
return self
|
|
|
|
def save(self):
|
|
raise NotImplementedError
|
|
|
|
def load(self):
|
|
raise NotImplementedError |