f8175e0221bbf31ffc8c10ee809b7ee2e118362d,oddt/scoring/functions/RFScore.py,rfscore,train,#rfscore#,93

Before Change

        print("Training RFScore v%i on PDBBind v%i"
              % (self.version, pdbbind_version), file=sys.stderr)

        error = rmse(self.model.predict(self.test_descs), self.test_target)
        r2 = self.model.score(self.test_descs, self.test_target)
        r = np.sqrt(r2)
        print("Test set:",
              "R**2: %.4f" % r2,

After Change

        sets = [
            ("Test", self.model.predict(self.test_descs), self.test_target),
            ("Train", self.model.predict(self.train_descs), self.train_target),
            ("OOB", self.model.oob_prediction_, self.train_target)]

        for name, pred, target in sets:
            print("%s set:" % name,

Instances