fix
This commit is contained in:
parent
14fc6d1120
commit
7e735543b9
10
model.py
10
model.py
@ -51,10 +51,14 @@ import tensorflow as tf
|
|||||||
|
|
||||||
data = pd.read_csv('./data/train.csv')
|
data = pd.read_csv('./data/train.csv')
|
||||||
|
|
||||||
features_idx = [1, 4, 7] # Sex, Age, BodyweightKg
|
print(data.columns) # Debugging: Print DataFrame columns
|
||||||
target_idx = 25 # TotalKg
|
|
||||||
|
|
||||||
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]
|
features = data.iloc[:, features_idx]
|
||||||
target = data.iloc[:, target_idx]
|
target = data.iloc[:, target_idx]
|
||||||
|
Loading…
Reference in New Issue
Block a user