NameError: имя 'X_train1' не определено

Question

Я выполняю код, передавая некоторые данные тестирования и обучения в функции, но когда я использую его в той же функции, в которой передаются данные, он выдает ошибку: NameError 'name' X_train1 'not определен'.Я использую Sypder IDE для выполнения кода.Вот код

def visualize_result(X_train1,y_train1,X_test1,y_test1,regressor):
    plt.scatter(X_train1, y_train1, color='red')
    plt.plot(X_train1, regressor.predict(X_train1), color='blue')
def main():
    visualize_result(train_X,train_Y,test_X,test_Y,regressor)

if __name__ == "__main__":
    main()