This commit is contained in:
Szymon Bartanowicz 2024-05-14 23:07:05 +02:00
parent c94695e20b
commit da08a94eb5

View File

@ -35,8 +35,8 @@ pipeline = Pipeline(steps=[
pipeline['model'].compile(optimizer='adam', loss='mse', metrics=['mae']) pipeline['model'].compile(optimizer='adam', loss='mse', metrics=['mae'])
X_train_excluded = X_train.iloc[:, 1:] X_train_excluded = X_train.iloc[1:, :]
y_train_excluded = y_train.iloc[:, 1:] y_train_excluded = y_train.iloc[1:, :]
pipeline.fit(X_train_excluded, y_train_excluded, model__epochs=10, model__validation_split=0.1) pipeline.fit(X_train_excluded, y_train_excluded, model__epochs=10, model__validation_split=0.1)