qxmt.visualization.plot_dataset module

qxmt.visualization.plot_dataset module#

qxmt.visualization.plot_dataset.plot_2d_dataset(dataset, colors=None, class_labels=None, save_path=None, **kwargs)

Ploat dataset on 2D plane.

Parameters:

  • dataset (Dataset) – Dataset object. It contains train and test data.

  • colors (Optional[dict[int, tuple]], optional) – color of each class. Defaults to None.

  • class_labels (Optional[dict[int, str]], optional) – label of each class. Defaults to None.

  • save_path (Optional[str], optional) – save path of graph. Defaults to None.

  • **kwargs (Any) – additional arguments for plot. train_title (str, optional): title of train dataset. Defaults to “Train Dataset”. test_title (str, optional): title of test dataset. Defaults to “Test Dataset”.

Return type:

None