s464937/ium_464937
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Szymon Bartanowicz 2024-05-14 23:43:14 +02:00
parent a597ba4d2a
commit 5651fb92d1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model.py
View File

@ -29,7 +29,7 @@ preprocessor = ColumnTransformer(
pipeline = Pipeline(steps=[ pipeline = Pipeline(steps=[
('preprocessor', preprocessor), ('preprocessor', preprocessor),
('model', Sequential([ ('model', Sequential([
Dense(64, activation='relu', input_dim=4), Dense(64, activation='relu', input_dim=5),
Dense(64, activation='relu'), Dense(64, activation='relu'),
Dense(1) Dense(1)
])) ]))