Załadowanie danych:

dataset = read.csv('Position_Salaries.csv')
dataset = dataset[2:3]

I nauczenie modelu. W przypadku R korzystamy z funkcji regresji liniowej, jednak dodajemy kolejne potęgi zmiennej niezależnej jako nowe zmienne niezależne:

ataset$Level2 = dataset$Level^2
dataset$Level3 = dataset$Level^3
dataset$Level4 = dataset$Level^4
poly_reg = lm(formula = Salary ~ .,
              data = dataset)