Несоответствие числа образцов ошибки из Python

Я работаю над конкурсом "Титаник" на IDE Spyder. Код едва закончен, но я делаю это один шаг за раз (и это первый раз, когда я когда-либо строил обучающую модель). Теперь я получаю Found input variables with inconsistent numbers of samples: [891, 183] ошибка Found input variables with inconsistent numbers of samples: [891, 183] в журнале при попытке запустить мой код. Это то, что у меня есть до сих пор:

import pandas as pd
from sklearn.tree import DecisionTreeRegressor
from sklearn.model_selection import train_test_split
from sklearn.metrics import mean_absolute_error

train_path = "C:\\Users\\Omar\\Downloads\\Titanic Data\\train.csv"
train_data = pd.read_csv(train_path)
columns_of_interest = ['Survived','Pclass', 'Sex', 'Age']
filtered_titanic_data = train_data.dropna(axis=0)

x = train_data[columns_of_interest]
y = filtered_titanic_data.Survived

train_x, val_x, train_y, val_y = train_test_split(x, y, random_state=0)

titanic_model = DecisionTreeRegressor()
titanic_model.fit(train_x, train_y)

val_predictions = titanic_model.predict(val_x)

print(filtered_titanic_data)

Idk ли это выход из файла excel или параметров. Извините, если это простой вопрос. Я не мог реализовывать решения других людей.