File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed
ensemble/_hist_gradient_boosting Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -39,7 +39,7 @@ class TreeNode:
39
39
----------
40
40
depth : int
41
41
The depth of the node, i.e. its distance from the root.
42
- sample_indices : ndarray of shape (n_samples_at_node,), dtype=np.uint
42
+ sample_indices : ndarray of shape (n_samples_at_node,), dtype=np.uint32
43
43
The indices of the samples at the node.
44
44
sum_gradients : float
45
45
The sum of the gradients of the samples at the node.
@@ -50,7 +50,7 @@ class TreeNode:
50
50
----------
51
51
depth : int
52
52
The depth of the node, i.e. its distance from the root.
53
- sample_indices : ndarray of shape (n_samples_at_node,), dtype=np.uint
53
+ sample_indices : ndarray of shape (n_samples_at_node,), dtype=np.uint32
54
54
The indices of the samples at the node.
55
55
sum_gradients : float
56
56
The sum of the gradients of the samples at the node.
Original file line number Diff line number Diff line change @@ -98,7 +98,7 @@ class KBinsDiscretizer(TransformerMixin, BaseEstimator):
98
98
The edges of each bin. Contain arrays of varying shapes ``(n_bins_, )``
99
99
Ignored features will have empty arrays.
100
100
101
- n_bins_ : ndarray of shape (n_features,), dtype=np.int_
101
+ n_bins_ : ndarray of shape (n_features,), dtype=np.int64
102
102
Number of bins per feature. Bins whose width are too small
103
103
(i.e., <= 1e-8) are removed with a warning.
104
104
@@ -439,7 +439,7 @@ def inverse_transform(self, Xt):
439
439
for jj in range (n_features ):
440
440
bin_edges = self .bin_edges_ [jj ]
441
441
bin_centers = (bin_edges [1 :] + bin_edges [:- 1 ]) * 0.5
442
- Xinv [:, jj ] = bin_centers [np . int_ (Xinv [:, jj ])]
442
+ Xinv [:, jj ] = bin_centers [(Xinv [:, jj ]). astype ( np . int64 )]
443
443
444
444
return Xinv
445
445
You can’t perform that action at this time.
0 commit comments