Skip to content

Linear Regression

Fit a least-squares line to each feature's training arrays on the GPU and predict a value.

Live Playground

Highlights

  • several array attributes in one function
  • scalar and array inputs mixed
  • one prediction per feature

This example comes from the autk-grammar repository.

Released under the MIT License.