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 22:29:02 +02:00
parent 14fc6d1120
commit 7e735543b9
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
model.py
View File

@ -51,10 +51,14 @@ import tensorflow as tf
data = pd.read_csv('./data/train.csv')
features_idx = [1, 4, 7] # Sex, Age, BodyweightKg
target_idx = 25 # TotalKg
print(data.columns) # Debugging: Print DataFrame columns
data = data.iloc[:, [1, 4, 7, 25]].dropna()
# Assuming the relevant columns are at these indexes
features_idx = [1, 4, 7] # Sex, Age, BodyweightKg
target_idx = 24 # TotalKg
# Dropping rows with NaN values from relevant columns
data = data.iloc[:, [1, 4, 7, 24]].dropna()
features = data.iloc[:, features_idx]
target = data.iloc[:, target_idx]