src.canns.analyzer.slow_points.visualization¶
Visualization functions for fixed point analysis.
Functions¶
|
Plot fixed points and trajectories in 2D using PCA. |
|
Plot fixed points and trajectories in 3D using PCA. |
Module Contents¶
- src.canns.analyzer.slow_points.visualization.plot_fixed_points_2d(fixed_points, state_traj, config=None, plot_batch_idx=None, plot_start_time=0)[source]¶
Plot fixed points and trajectories in 2D using PCA.
- Parameters:
fixed_points (src.canns.analyzer.slow_points.fixed_points.FixedPoints) – FixedPoints object containing analysis results.
state_traj (numpy.ndarray) – State trajectories [n_batch x n_time x n_states].
config (src.canns.analyzer.plotting.config.PlotConfig | None) – Plot configuration. If None, uses default config.
plot_batch_idx (list[int] | None) – Batch indices to plot trajectories. If None, plots first 30.
plot_start_time (int) – Starting time index for trajectory plotting.
- Returns:
matplotlib Figure object.
- Return type:
matplotlib.figure.Figure
Example
>>> from canns.analyzer.slow_points import plot_fixed_points_2d, FixedPoints >>> from canns.analyzer.plotting import PlotConfig >>> config = PlotConfig( ... title="Fixed Points Analysis", ... figsize=(10, 8), ... save_path="fps_2d.png" ... ) >>> fig = plot_fixed_points_2d(unique_fps, hiddens, config=config)
- src.canns.analyzer.slow_points.visualization.plot_fixed_points_3d(fixed_points, state_traj, config=None, plot_batch_idx=None, plot_start_time=0)[source]¶
Plot fixed points and trajectories in 3D using PCA.
- Parameters:
fixed_points (src.canns.analyzer.slow_points.fixed_points.FixedPoints) – FixedPoints object containing analysis results.
state_traj (numpy.ndarray) – State trajectories [n_batch x n_time x n_states].
config (src.canns.analyzer.plotting.config.PlotConfig | None) – Plot configuration. If None, uses default config.
plot_batch_idx (list[int] | None) – Batch indices to plot trajectories. If None, plots first 30.
plot_start_time (int) – Starting time index for trajectory plotting.
- Returns:
matplotlib Figure object.
- Return type:
matplotlib.figure.Figure
Example
>>> from canns.analyzer.slow_points import plot_fixed_points_3d, FixedPoints >>> from canns.analyzer.plotting import PlotConfig >>> config = PlotConfig( ... title="Fixed Points 3D", ... figsize=(12, 10), ... save_path="fps_3d.png" ... ) >>> fig = plot_fixed_points_3d(unique_fps, hiddens, config=config)