Beispiel #1
0
def test_custom_metric_computation_multinomial():
    (model, f_test) = multinomial_model(H2OGradientBoostingEstimator,
                                        custom_rmse_mm())
    assert_correct_custom_metric(model, f_test, "rmse", "Multinomial on iris")
Beispiel #2
0
def test_custom_metric_computation_logloss():
    (model, f_test) = binomial_model(H2OGradientBoostingEstimator,
                                     custom_logloss_mm())
    assert_correct_custom_metric(model, f_test, "logloss",
                                 "Logloss on prostate")
Beispiel #3
0
def test_custom_metric_computation_binomial():
    (model, f_test) = binomial_model(H2OGradientBoostingEstimator,
                                     custom_rmse_mm())
    assert_correct_custom_metric(model, f_test, "rmse", "Binomial on prostate")
Beispiel #4
0
def test_custom_metric_computation_regression():
    (model, f_test) = regression_model(H2OGradientBoostingEstimator,
                                       custom_mae_mm())
    assert_correct_custom_metric(model, f_test, "mae",
                                 "Regression on prostate")
def test_custom_metric_computation_multinomial():
    (model, f_test) = multinomial_model(H2ORandomForestEstimator, custom_rmse_mm())
    assert_correct_custom_metric(model, f_test, "rmse", "Multinomial on iris")
def test_custom_metric_computation_binomial():
    (model, f_test) = binomial_model(H2ORandomForestEstimator, custom_rmse_mm())
    assert_correct_custom_metric(model, f_test, "rmse", "Binomial on prostate")
def test_custom_metric_computation_regression():
    (model, f_test) = regression_model(H2ORandomForestEstimator, custom_mae_mm())
    assert_correct_custom_metric(model, f_test, "mae", "Regression on prostate")