replay_trajectory_classification.continuous_state_transitions.RandomWalk#

class RandomWalk(environment_name: str = '', movement_var: float = 6.0, movement_mean: float = 0.0, use_diffusion: bool = False)[source]#

Bases: object

A transition where the movement stays locally close in space

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

movement_mean#
Type:

float, optional

use_diffusion#

Use diffusion to respect the geometry of the environment

Type:

bool, 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_mean

movement_var

use_diffusion

make_state_transition(environments: tuple[Environment]) → ndarray[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)