File tree Expand file tree Collapse file tree 2 files changed +0
-5
lines changed
hidimstat/estimator/tests Expand file tree Collapse file tree 2 files changed +0
-5
lines changed Original file line number Diff line number Diff line change @@ -29,8 +29,6 @@ def generate_data(
2929 Data matrix
3030 y : np.array
3131 Target vector
32- grps : np.array
33- Group vector
3432 """
3533 rng = np .random .default_rng (seed )
3634 if problem_type == "regression" :
@@ -48,7 +46,6 @@ def generate_data(
4846 n_features = n_features ,
4947 random_state = seed ,
5048 )
51- # y = np.array([str(i) for i in y])
5249
5350 X = pd .DataFrame (X , columns = [f"col{ i + 1 } " for i in range (n_features )])
5451
Original file line number Diff line number Diff line change 99def test_RandomForestRegressorModified ():
1010 """
1111 Test the RandomForestRegressorModified for regression.
12- Parameters:
13- - regression_data: A tuple containing the input features (X) and target variable (y) for regression.
1412 """
1513 X , y = generate_data (problem_type = "regression" )
1614 learner = RandomForestRegressorModified (n_jobs = 10 , verbose = 0 )
You can’t perform that action at this time.
0 commit comments