sklearn实战:使用线性回归算法拟合正弦函数

%matplotlib inline
import matplotlib.pyplot as plt
import numpy as np
n_dots = 200

X = np.linspace(-2 * np.pi, 2 * np.pi, n_dots)
Y = np.sin(X) + 0.2 * np.random.rand(n_dots) - 0.1
X = X.reshape(-1, 1)
Y = Y.reshape(-1, 1)

使用sklearn.preprocessing.PolynomialFeatures来进行特征的构造。

它是使用多项式的方法来进行的,如果有a,b两个特征,那么它的2次多项式为(1,a,b,a^2,ab, b^2)。

Polyn

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要实现拟合分段函数的多元线性回归,可以通过引入分段函数的指示变量来实现。具体来说,可以将自变量按照分段点进行分段,然后对于每个分段引入一个指示变量,表示该自变量是否在该分段内。 例如,假设有两个自变量 X1 和 X2,要拟合两个分段的分段函数,可以将 X1 和 X2 分别按照分段点进行分段,得到四个区间。然后引入四个指示变量,分别表示 X1 和 X2 是否在每个区间内。这样,就可以将分段函数转化为多元线性回归的形式。 在 sklearn 中,可以使用 PolynomialFeatures 类来进行多项式特征转换,将自变量转化为多项式特征,并引入指示变量。然后使用 LinearRegression 类进行拟合。具体实现可以参考以下示例代码: ```python from sklearn.preprocessing import PolynomialFeatures from sklearn.linear_model import LinearRegression # 定义分段点 segment_points = [1, 2] # 生成多项式特征 poly = PolynomialFeatures(degree=1, include_bias=False) X_poly = poly.fit_transform(X) # 引入指示变量 X_indicator = np.zeros((X.shape[0], len(segment_points) + 1)) for i, p in enumerate(segment_points): X_indicator[:, i] = (X[:, 0] >= p) X_indicator[:, -1] = 1 # 拼接多项式特征和指示变量 X_new = np.hstack((X_poly, X_indicator)) # 拟合线性回归模型 reg = LinearRegression().fit(X_new, y) ``` 其中,X 是自变量的样本数据,y 是因变量的样本数据。segment_points 是分段点的列表,degree 是多项式特征的次数。最终得到的 X_new 包含了多项式特征和指示变量,可以用于拟合线性回归模型。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值