xnie / rlearner

Quasi-Oracle Estimation of Heterogeneous Treatment Effects

Home Page:https://arxiv.org/abs/1712.04912

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Error in UseMethod("predict") : no applicable method for 'predict' applied to an object of class "tlearner"

dbroockman opened this issue · comments

The ReadMe has the following example:

tlearner_fit = tlearner_cv(data$x, data$w, data$y, model_specs=model_specs)
tlearner_tau_hat = predict(tlearner_fit, data$x)

However, when I run this code, I see the following error:

Error in UseMethod("predict") : 
  no applicable method for 'predict' applied to an object of class "tlearner"

Thanks for bringing this to our attention! Due to version issues and package limitations, we are no longer supporting caret-based estimators, and are planning on updating the repository to reflect this change. However, this does not affect replicating the results from the paper as the paper does not use caret-based code.