nntoolbox.visualization.reduce_dim module

nntoolbox.visualization.reduce_dim.visualize_data(data: numpy.ndarray, labels: numpy.ndarray, method)

Reduce dimension of data to 2D and visualize using a method

Parameters

• data – data. a 2D numpy array (batch_size, dimension)

• labels – labels of data (for coloring)

• method – a method (e.g PCA). Should be a sklearn class

nntoolbox.visualization.reduce_dim.visualize_mds(data, labels)

nntoolbox.visualization.reduce_dim.visualize_pca(data, labels)

nntoolbox.visualization.reduce_dim.visualize_tsne(data, labels)