Commit e4e75dc0 authored by Tiphagne Thomas's avatar Tiphagne Thomas
Browse files

Test merge

parent f355aa64
print("hello world")
import h2o
from h2o.automl import H2OAutoML
h2o.init()
train = h2o.import_file("https://s3.amazonaws.com/erin-data/higgs/higgs_train_10k.csv")
test = h2o.import_file("https://s3.amazonaws.com/erin-data/higgs/higgs_test_5k.csv")
x = train.columns
y = "response"
x.remove(y)
train[y] = train[y].asfactor()
test[y] = test[y].asfactor()
aml = H2OAutoML(max_runtime_secs = 130)
aml.train(x = x, y = y, training_frame = train, leaderboard_frame = test)
print(aml.leaderboard)
print(aml.leader)
print(aml.predict(test))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment