Skip to content

Commit ab74deb

Browse files
Update values.py
1 parent de4fe2c commit ab74deb

1 file changed

Lines changed: 12 additions & 10 deletions

File tree

src/spotPython/hyperparameters/values.py

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -610,9 +610,11 @@ def replace_levels_with_positions(hyper_dict, hyper_dict_values) -> dict:
610610
dictionary with levels
611611
hyper_dict_values (dict):
612612
dictionary with values
613+
613614
Returns:
614615
(dict):
615616
dictionary with values
617+
616618
Examples:
617619
>>> from spotPython.hyperparameters.values import replace_levels_with_positions
618620
hyper_dict = {"leaf_prediction": {
@@ -819,16 +821,16 @@ def get_one_sklearn_model_from_X(X, fun_control=None):
819821
The sklearn model.
820822
821823
Examples:
822-
>>> from sklearn.linear_model import LinearRegression
823-
from spotRiver.data.sklearn_hyper_dict import SklearnHyperDict
824-
fun_control = {}
825-
add_core_model_to_fun_control(core_model=LinearRegression,
826-
fun_control=func_control,
827-
hyper_dict=SklearnHyperDict,
828-
filename=None)
829-
X = np.array([0, 0, 0, 0, 0])
830-
get_one_sklearn_model_from_X(X, fun_control)
831-
LinearRegression()
824+
>>> from sklearn.linear_model import LinearRegression
825+
from spotRiver.data.sklearn_hyper_dict import SklearnHyperDict
826+
fun_control = {}
827+
add_core_model_to_fun_control(core_model=LinearRegression,
828+
fun_control=func_control,
829+
hyper_dict=SklearnHyperDict,
830+
filename=None)
831+
X = np.array([0, 0, 0, 0, 0])
832+
get_one_sklearn_model_from_X(X, fun_control)
833+
LinearRegression()
832834
"""
833835
core_model = get_one_core_model_from_X(X=X, fun_control=fun_control)
834836
if fun_control["prep_model"] is not None:

0 commit comments

Comments
 (0)