def save(self):
Tools.create_dir_if_not_exists(self.save_directory)
joblib.dump(self.umap_fit,
os.path.join(self.save_directory, self.umap_save_name))
joblib.dump(self.dbscan_fit,
os.path.join(self.save_directory, self.dbscan_save_name))
def save(self):
Tools.create_dir_if_not_exists(self.save_directory)
joblib.dump(self.pca_fit,
os.path.join(self.save_directory, self.pca_save_name))
joblib.dump(self.kmeans_fit,
os.path.join(self.save_directory, self.kmeans_save_name))
joblib.dump(self.cah_fit,
os.path.join(self.save_directory, self.cah_save_name))
def save(self):
Tools.create_dir_if_not_exists(self.config['save_directory'])
joblib.dump(
self.pca_fit,
os.path.join(self.config['save_directory'], self.pca_save_name))
joblib.dump(
self.kmeans_fit,
os.path.join(self.config['save_directory'], self.kmeans_save_name))
joblib.dump(
self.dbscan_fit,
os.path.join(self.config['save_directory'], self.dbscan_save_name))
def __init__(self, config, pictures_id, pictures_np):
self.config = config
self.save_directory = os.path.join(
self.config['save_directory'], '%s_%s_%s' %
(self.config['model']['type'], self.config['model']['name'],
self.config['version']))
self.pictures_id = pictures_id
self.pictures_np = pictures_np
self.final_labels = None
self.colors = None
Tools.create_dir_if_not_exists(self.save_directory)