fix
This commit is contained in:
parent
a188435a1b
commit
97ca282402
2
model.py
2
model.py
@ -11,7 +11,7 @@ import tensorflow as tf
|
||||
data = pd.read_csv('./data/train.csv')
|
||||
|
||||
data = data[['Sex', 'Age', 'BodyweightKg', 'TotalKg']].dropna()
|
||||
data['Age'] = data['Age'].astype(int)
|
||||
data['Age'] = pd.to_numeric(data['Age'], errors='coerce')
|
||||
features = data[['Sex', 'Age', 'BodyweightKg']]
|
||||
target = data['TotalKg']
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user