Importance Class Reference

Inheritance diagram for Importance:

Public Member Functions
def	add (self, data, columns, variables)
def	finish (self)

Detailed Description

Plots importance matrix

Definition at line 1185 of file plotting.py.

Member Function Documentation

add()

def add	(		self,
			data,
			columns,
			variables
	)

Add a new correlation plot.
@param data pandas.DataFrame containing all data
@param columns which are used to calculate the correlations

Reimplemented from Plotter.

Definition at line 1190 of file plotting.py.

    def add(self, data, columns, variables):
        """
        Add a new correlation plot.
        @param data pandas.DataFrame containing all data
        @param columns which are used to calculate the correlations
        """
        self.figure.set_tight_layout(True)
 
        def norm(x):
            width = (numpy.max(x) - numpy.min(x))
            if width <= 0:
                return numpy.zeros(x.shape)
            return (x - numpy.min(x)) / width * 100
 
        importance_matrix = numpy.vstack([norm(data[column]) for column in columns]).T
        importance_heatmap = self.axis.pcolor(importance_matrix, cmap=plt.cm.RdBu, vmin=0.0, vmax=100,
                                              rasterized=True)
 
        # put the major ticks at the middle of each cell
        self.axis.set_yticks(numpy.arange(importance_matrix.shape[0]) + 0.5, minor=False)
        self.axis.set_xticks(numpy.arange(importance_matrix.shape[1]) + 0.5, minor=False)
 
        self.axis.set_xticklabels(columns, minor=False, rotation=90)
        self.axis.set_yticklabels(variables, minor=False)
 
        self.axis.xaxis.tick_top()
 
        for y in range(importance_matrix.shape[0]):
            for x in range(importance_matrix.shape[1]):
                txt = self.axis.text(x + 0.5, y + 0.5, f'{importance_matrix[y, x]:.0f}',
                                     size=14,
                                     horizontalalignment='center',
                                     verticalalignment='center',
                                     color='w')
                txt.set_path_effects([PathEffects.withStroke(linewidth=3, foreground='k')])
 
        cb = self.figure.colorbar(importance_heatmap, ticks=[0.0, 100], orientation='vertical')
        cb.ax.set_yticklabels(['low', 'high'])
 
        # remove whitespace
        self.axis.set_ylim(0, importance_matrix.shape[0])
 
        self.axis.set_aspect('equal')
 
        return self
 

finish()

def finish

(

self

)

Sets limits, title, axis-labels and legend of the plot

Reimplemented from Plotter.

Definition at line 1236 of file plotting.py.

    def finish(self):
        """
        Sets limits, title, axis-labels and legend of the plot
        """
        return self
 
 

---

The documentation for this class was generated from the following file:

• mva/scripts/basf2_mva_evaluation/plotting.py