loss_history, weight_history = linear_regression(X_train, y_train,