-
Notifications
You must be signed in to change notification settings - Fork 96
/
linear_regressor.py
28 lines (25 loc) · 1.52 KB
/
linear_regressor.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
from ..common._registration import register_shape_calculator
from ..common.shape_calculator import calculate_linear_regressor_output_shapes
register_shape_calculator('SklearnAdaBoostRegressor',
calculate_linear_regressor_output_shapes)
register_shape_calculator('SklearnLinearRegressor',
calculate_linear_regressor_output_shapes)
register_shape_calculator('SklearnLinearSVR',
calculate_linear_regressor_output_shapes)
register_shape_calculator('SklearnDecisionTreeRegressor',
calculate_linear_regressor_output_shapes)
register_shape_calculator('SklearnRandomForestRegressor',
calculate_linear_regressor_output_shapes)
register_shape_calculator('SklearnExtraTreesRegressor',
calculate_linear_regressor_output_shapes)
register_shape_calculator('SklearnGradientBoostingRegressor',
calculate_linear_regressor_output_shapes)
register_shape_calculator('SklearnKNeighborsRegressor',
calculate_linear_regressor_output_shapes)
register_shape_calculator('SklearnMLPRegressor',
calculate_linear_regressor_output_shapes)