gradient_descent: fit a linear model

In carolineying/bis557: BIS 557 Assignment

fit a linear model

Description

This is a function taking a formula, a dataframe and an optional list of contrast for factor variables, and using gradient descent to return to coefficients. with reference to https://towardsdatascience.com/linear-regression-using-gradient-descent-97a6c8700931

Usage

gradient_descent(formula, df, contrasts = NULL, gamma = 1e-04, iter = 10^5)

Arguments

formula	an object of class "formula": describing the model to be fitted.
df	a dataframe that containing all the variables in the model.
contrasts	an optional list of contrasts for factor variables.
gamma	gamma_k, the learning rate
iter	number of iterations

Examples

data(iris)
form <- Sepal.Length ~ Sepal.Width
gradient_descent(formula = form,df = iris)

carolineying/bis557 documentation built on April 17, 2023, 4 a.m.