From f39bf6491530f6a17b7be24f8d691f218f94bccd Mon Sep 17 00:00:00 2001 From: Szymon Bartanowicz Date: Tue, 14 May 2024 23:58:43 +0200 Subject: [PATCH] fix --- model.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/model.py b/model.py index aedcc83..389327e 100644 --- a/model.py +++ b/model.py @@ -11,7 +11,9 @@ import tensorflow as tf data = pd.read_csv('./data/train.csv') data = data[['Sex', 'Age', 'BodyweightKg', 'TotalKg']].dropna() - +data['Age'] = pd.to_numeric(data['Age'], errors='coerce') +data['BodyweightKg'] = pd.to_numeric(data['BodyweightKg'], errors='coerce') +data['TotalKg'] = pd.to_numeric(data['TotalKg'], errors='coerce') features = data[['Sex', 'Age', 'BodyweightKg']] target = data['TotalKg'] @@ -27,7 +29,7 @@ preprocessor = ColumnTransformer( pipeline = Pipeline(steps=[ ('preprocessor', preprocessor), ('model', Sequential([ - Dense(64, activation='relu', input_dim=4), + Dense(64, activation='relu', input_dim=5), Dense(64, activation='relu'), Dense(1) ]))