Рамка H2O пуста - PullRequest
       25

Рамка H2O пуста

0 голосов
/ 26 сентября 2019

Я пытаюсь запустить пример кода, который я нашел по ссылке ниже.

http://docs.h2o.ai/h2o/latest-stable/h2o-docs/automl.html#automl-interface

Я немного изменил код.Мой фактический код приведен ниже.

import h2o
from h2o.automl import H2OAutoML
import pandas as pd
from sklearn.model_selection import train_test_split

h2o.init()

df = pd.read_csv('C:\\my_path\\taxi.csv', header=0, encoding = 'unicode_escape')
pd.set_option('display.max_columns', None)
df.head(10)

df = df[:100000]

y = df.fare_amount

# create training and testing vars
X_train, X_test, y_train, y_test = train_test_split(df, y, test_size=0.2)

# Identify predictors and response
y = df['fare_amount']
del df['fare_amount']
x = df

# Run AutoML for 20 base models (limited to 1 hour max runtime by default)
train = h2o.H2OFrame(df)

aml = H2OAutoML(max_models=20, seed=1)
aml.train(x=x, y=y, training_frame=train)

# View the AutoML Leaderboard
lb = aml.leaderboard
lb.head(rows=lb.nrows)

Я установил в качестве целевой переменной переменную тарифа.Когда я запускаю код, все, что я получаю, это сообщение: «Этот H2OFrame пустой».Я могу упустить что-то довольно простое, но я не вижу, что это такое.Любые мысли, кто-нибудь?

...