Linear Regression

Fit a simple linear regression and return the formula. Optionally, use one or more group_by columns to create a regression per unique grouping.
Currently, only supports a single independent variable.
Parameters
Name
Type
Description
Is Optional
group_by
column_list
Columns to group by before building the linear regression model. Use this field to create multiple models (one per unique grouping)
True
y
column
Dependent variable for the linear regression
​
x
column
Independent variable for the linear regression
​
Example
internet_sales = rasgo.get.dataset(74)
​
ds1 = internet_sales.aggregate(
    group_by=['PRODUCTKEY','CUSTOMERKEY'],
    aggregations={'SALESAMOUNT':['AVG'],
                'TOTALPRODUCTCOST':['AVG']})
​
ds2 = ds1.linear_regression(
  x = 'SALESAMOUNT_AVG',
  y = 'TOTALPRODUCTCOST_AVG')
​
ds2.preview()
Source Code
RasgoTransforms/linear_regression.sql at main · rasgointelligence/RasgoTransforms
GitHub

Name	Type	Description	Is Optional
group_by	column_list	Columns to group by before building the linear regression model. Use this field to create multiple models (one per unique grouping)	True
y	column	Dependent variable for the linear regression
x	column	Independent variable for the linear regression

Lead

Market Basket

Last modified 1yr ago