qxmt.visualization.plot_dataset module

qxmt.visualization.plot_dataset module#

qxmt.visualization.plot_dataset.plot_2d_dataset(dataset, colors=None, class_labels=None, train_title='Train Dataset', test_title='Test Dataset', 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.

  • train_title (str, optional) – title of train dataset. Defaults to “Train Dataset”.

  • test_title (str, optional) – title of test dataset. Defaults to “Test Dataset”.

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

  • **kwargs (Any) – additional arguments for scatter plot.

Return type:

None