def generateClusteringVisualization(self):
if self.families_analysis:
clustering = Clustering(self.categories.instances,
self.categories.assigned_categories)
clustering.generateClustering(None, None)
else:
self.clustering_exp = None
def generateClusteringVisualization(self):
if self.families_analysis:
self.clustering_exp = self.createClusteringExperiment()
clustering = Clustering(self.categories.instances,
self.categories.assigned_categories)
clustering.generateClustering(None, None)
clustering.export(self.clustering_exp.getOutputDirectory())
else:
self.clustering_exp = None
def alertsClassification(self, alerts_ids):
multiclass_model = self.buildMulticlassClassifier(alerts_ids)
num_families = len(
self.datasets.train_instances.getFamiliesValues(label='malicious'))
clustering_experiment = self.createClusteringExperiment(
num_clusters=num_families)
self.grouping_exp_id = clustering_experiment.experiment_id
all_families = list(multiclass_model.class_labels)
predicted_families = multiclass_model.testing_monitoring.getPredictedLabels(
)
predicted_families = [
all_families.index(x) for x in predicted_families
]
clustering = Clustering(clustering_experiment,
multiclass_model.datasets.test_instances,
predicted_families)
clustering.generateClustering(None, None, cluster_labels=all_families)
