Как получить координаты кривой точного возврата? - PullRequest
Новая
       5

Как получить координаты кривой точного возврата?

0 голосов
/ 08 февраля 2020

Я пытаюсь получить координаты для кривой точности и отзыва для модели, чтобы я мог построить их график.

Я могу получить координаты кривой ro c через: 

cars = h2o.import_file("https://s3.amazonaws.com/h2o-public-test-data/smalldata/junit/cars_20mpg.csv")
cars["economy_20mpg"] = cars["economy_20mpg"].asfactor()
r = cars[0].runif()
train = cars[r > .2]
valid = cars[r <=.2]
response_col = "economy_20mpg"
distribution = "bernoulli"
predictors = ["displacement", "power", "weight", "acceleration", "year"]
from h2o.estimators.gbm import H2OGradientBoostingEstimator
gbm = H2OGradientBoostingEstimator(nfolds=3,
                                   distribution=distribution,
                                   fold_assignment="Random")
gbm.train(y=response_col,
          x=predictors,
          validation_frame=valid,
          training_frame=train)
print(gbm.roc())

но gbm.recall () - это просто пороговое значение и одно значение.

...