replay_trajectory_classification.continuous_state_transitions.Uniform#

class Uniform(environment_name: str = '', environment2_name: str | None = None)[source]#

Bases: object

environment_name#

Name of first environment to fit

Type:

str, optional

environment_name2#

Name of second environment to fit if going from one environment to another

Type:

str, optional

Attributes:
environment2_name

Methods

make_state_transition(environments)

Creates a transition matrix for a given environment.

Methods

make_state_transition

Creates a transition matrix for a given environment.

Attributes

environment2_name

environment_name

make_state_transition(environments: tuple[Environment])[source]#

Creates a transition matrix for a given environment.

Parameters:

environments (tuple[Environment]) – The existing environments in the model

Returns:

state_transition_matrix

Return type:

np.ndarray, shape (n_position_bins, n_position_bins)