movement_primitives.data.generate_1d_trajectory_distribution

movement_primitives.data.generate_1d_trajectory_distribution(n_demos, n_steps, initial_offset_range=3.0, final_offset_range=0.1, noise_per_step_range=20.0, random_state=RandomState(MT19937) at 0x7F037D3E4540)

Generates toy data for testing and demonstration.

Parameters:

n_demosint

Number of demonstrations

n_stepsint

Number of steps

initial_offset_rangefloat, optional (default: 3)

Range of initial offset from cosine

final_offset_rangefloat, optional (default: 0.1)

Range of final offset from cosine

noise_per_step_rangefloat, optional (default: 20)

Factor for noise in each step

random_stateRandomState, optional (default: seed 0)

Random state

Returns:

Tarray, shape (n_steps,)

Times

Yarray, shape (n_demos, n_steps, 1)

Demonstrations (positions)