mlpy.mdp.discrete.DecisionTreeModel.fit

DecisionTreeModel.fit(obs, actions, rewards=None)[source]

Fit the model to the trajectory data.

Parameters:

obs : array_like, shape (nfeatures, n)

Trajectory of observations, where each observation has nfeatures features and n is the length of the trajectory.

actions : array_like, shape (nfeatures, n)

Trajectory of actions, where each action has nfeatures features and n is the length of the trajectory.

rewards : array_like, shape (n,)

List of rewards, a reward is awarded for each observation.