1. 程式人生 > >Python之嶺回歸

Python之嶺回歸

ict port randn mod alpha near python .com auth

實現:
# -*- coding: UTF-8 -*-

import numpy as np
from sklearn.linear_model import Ridge

__author__ = ‘zhen‘

X = 2 * np.random.rand(100, 1)
y = 4 + 3 * X + np.random.randn(100, 1)
# 嶺回歸
ridge_reg = Ridge(alpha=1, solver=‘sag‘)
ridge_reg.fit(X, y)
print("="*50)
print(ridge_reg.predict(1.5))
print(ridge_reg.intercept_)
print(ridge_reg.coef_)
print("="*50)
結果:
技術分享圖片

 

Python之嶺回歸