dask_ml.linear_model.LinearRegression
dask_ml.linear_model
.LinearRegression¶
- class dask_ml.linear_model.LinearRegression(penalty='l2', dual=False, tol=0.0001, C=1.0, fit_intercept=True, intercept_scaling=1.0, class_weight=None, random_state=None, solver='admm', max_iter=100, multi_class='ovr', verbose=0, warm_start=False, n_jobs=1, solver_kwargs=None)¶
线性回归的估计器。
- 参数
- penaltystr 或 Regularizer,默认为 ‘l2’
使用的正则化器。仅与 ‘admm’、‘lbfgs’ 和 ‘proximal_grad’ 求解器相关。
对于字符串值,只有 ‘l1’ 或 ‘l2’ 有效。
- dualbool
忽略
- tolfloat,默认为 1e-4
收敛容差。
- Cfloat
正则化强度。请注意,
dask-glm
求解器使用参数化 \(\lambda = 1 / C\)- fit_interceptbool,默认为 True
指定是否应在决策函数中添加一个常数(又称偏差或截距)。
- intercept_scalingbool
忽略
- class_weightdict 或 ‘balanced’
忽略
- random_stateint、RandomState 或 None
用于数据洗牌时伪随机数生成器的种子。如果为 int,random_state 是随机数生成器使用的种子;如果为 RandomState 实例,random_state 是随机数生成器;如果为 None,随机数生成器是 np.random 使用的 RandomState 实例。在 solver == ‘sag’ 或 ‘liblinear’ 时使用。
- solver{‘admm’, ‘gradient_descent’, ‘newton’, ‘lbfgs’, ‘proximal_grad’}
使用的求解器。详情请参阅 算法
- max_iterint,默认为 100
求解器收敛的最大迭代次数。
- multi_classstr,默认为 ‘ovr’
忽略。目前不支持多类别求解器。
- verboseint,默认为 0
忽略
- warm_startbool,默认为 False
忽略
- n_jobsint,默认为 1
忽略
- solver_kwargsdict,可选,默认为 None
要传递给求解器的额外关键字参数。
- 属性
- coef_数组,形状为 (n_classes, n_features)
模型系数的学习值
- intercept_float 或 None
截距的学习值(如果模型添加了截距)
示例
>>> from dask_glm.datasets import make_regression >>> X, y = make_regression() >>> lr = LinearRegression() >>> lr.fit(X, y) >>> lr.predict(X) >>> lr.predict(X) >>> lr.score(X, y)
方法
fit
(X[, y])在训练数据上拟合模型
get_metadata_routing
()获取此对象的元数据路由。
get_params
([deep])获取此估计器的参数。
predict
(X)预测 X 中样本的值。
score
(X, y)返回预测的决定系数 R^2。
set_params
(**params)设置此估计器的参数。
- __init__(penalty='l2', dual=False, tol=0.0001, C=1.0, fit_intercept=True, intercept_scaling=1.0, class_weight=None, random_state=None, solver='admm', max_iter=100, multi_class='ovr', verbose=0, warm_start=False, n_jobs=1, solver_kwargs=None)¶