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.