replay_trajectory_classification.continuous_state_transitions.RandomWalkDirection1#

class RandomWalkDirection1(environment_name: str = '', movement_var: float = 6.0)[source]#

Bases: object

A Gaussian random walk in that can only go one direction

environment_name#

Name of environment to fit

Type:

str, optional

movement_var#

How far the animal is can move in one time bin during normal movement.

Type:

float, optional

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

environment_name

movement_var

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)