ElasticNet

- 선형 모델의 한 종류로 Lasso의 L1과 Ridge의 L2 규제 방법을 결합

- 파라미터 alpha와 l1_ratio(L1 규제의 비율)를 이용해 규제의 정도 조정

- 2가지 방식을 결합하기에 시간이 오래 걸린다는 단점이 존재함


예제

from sklearn.linear_model import ElasticNet

model = ElasticNet(alpha = 1)
model.fit(x_train, y_train)
pred = model.predict(x_test)
mean_squared_error(y_test, pred)


alpha = 0.0001, l1_ratio = 0.6

from sklearn.linear_model import ElasticNet

model = ElasticNet(alpha = 0.0001, l1_ratio = 0.6)
model.fit(x_train, y_train)
pred = model.predict(x_test)
mean_squared_error(y_test, pred)

0개의 댓글