gradient_descent: Gradient Descent Function

Description Usage Arguments Value Author(s) Examples

Description

Implement gradient descent for ordinary least squares.

Usage

1
2
3
4
5
6
7
8
gradient_descent(
  form,
  d,
  contrasts = NULL,
  iters = 1e+05,
  gamma = 1e-04,
  threshold = 1e-12
)

Arguments

form

A formula

d

A dataframe

contrasts

Contrasts argument

iters

Number of maximum iterations

gamma

The learning rate

threshold

Threshold for the difference between RSS of new and old betas.

Value

a list of beta coefficients

Author(s)

Yijun Yang

Examples

1
2
3
4
## Not run: 
gradient_descent(body_mass_g ~ bill_depth_mm, penguins)

## End(Not run)

yijunyang/bis557 documentation built on Dec. 21, 2020, 3:06 a.m.