示例#1
0
    def CreatePlot(self):
        clusters = 1 if self.model.clusteringAlgorithm == 0 else self.model.clusters

        print('DEBUG - Clusters a graficar: %d' % clusters)
        plotter = Plotter()

        if (self.model.selectedAxes < 3):
            plotter.PlotSamples2D(self.model.dataset,
                                  axes=self.model.colsForAxes,
                                  labels=self.result.labels,
                                  clusters=clusters)

            if (len(self.result.centroids)):
                plotter.PlotCentroids2D(self.result.centroids,
                                        axes=self.model.colsForAxes)

        else:
            plotter.PlotSamples3D(self.model.dataset,
                                  axes=self.model.colsForAxes,
                                  labels=self.result.labels,
                                  clusters=clusters)

            if (len(self.result.centroids)):
                plotter.PlotCentroids3D(self.result.centroids,
                                        axes=self.model.colsForAxes)

        return plotter