h2oai / h2o-3

H2O is an Open Source, Distributed, Fast & Scalable Machine Learning Platform: Deep Learning, Gradient Boosting (GBM) & XGBoost, Random Forest, Generalized Linear Modeling (GLM with Elastic Net), K-Means, PCA, Generalized Additive Models (GAM), RuleFit, Support Vector Machine (SVM), Stacked Ensembles, Automatic Machine Learning (AutoML), etc.

Apache License 2.0

6.92k stars 2k forks source link

!Screen Shot 2018-06-19 at 3.32.33 PM.png|thumbnail!

{code:java} import h2o from h2o.estimators.gbm import H2OGradientBoostingEstimator import pandas import sys h2o.init(strict_version_check=False)

ecology = h2o.import_file("https://s3.amazonaws.com/h2o-public-test-data/smalldata/gbm_test/ecology_model.csv") ecology['Angaus'] = ecology['Angaus'].asfactor()

Set the predictors and the response column name

response = 'Angaus' predictors = ecology.columns[3:13]

gbm_model = H2OGradientBoostingEstimator(ntrees=50, learn_rate=0.05) gbm_model.train(x=predictors, y=response, training_frame=ecology)

pdp1=gbm_model.partial_plot(data=ecology,cols=['SegSumT'],server=False, plot=False,nbins=30) pdp1 pdp1.as_data_frame() {code}

h2oai / h2o-3

h2o python: when try to print the pdp table as data frame get error #12530

Set the predictors and the response column name