So I added a function where you can calculate a score for both the first stage and final stage models with any sklearn metric, or even any function that you pass. This will help enable more use of DML in industry where the data quality is low and the model performance is hard to interpret.