Method predict

rl2/mountaincar/n_step.py:39–40

(self, X)

    self.w += self.lr*(Y - X.dot(self.w)).dot(X)

  def predict(self, X):
    return X.dot(self.w)

# replace SKLearn Regressor
q_learning.SGDRegressor = SGDRegressor