loss_history, weight_history, bias_history = gradient_descent(X_train, y_train,