File tree 1 file changed +3
-11
lines changed 1 file changed +3
-11
lines changed Original file line number Diff line number Diff line change 33
33
from sklearn .externals .funcsigs import signature
34
34
35
35
36
- def _squeeze (length_scale ):
37
- """
38
- np.squeeze that returns a scalar instead of a 0-D array.
39
- """
40
- length_scale = np .squeeze (length_scale ).astype (float )
41
- if length_scale .ndim == 0 :
42
- length_scale = np .asscalar (length_scale )
43
- return length_scale
44
-
45
36
def _check_length_scale (X , length_scale ):
46
37
length_scale = np .squeeze (length_scale ).astype (float )
47
38
if np .ndim (length_scale ) > 1 :
@@ -1213,7 +1204,7 @@ def __repr__(self):
1213
1204
self .length_scale )))
1214
1205
else : # isotropic
1215
1206
return "{0}(length_scale={1:.3g})" .format (
1216
- self .__class__ .__name__ , _squeeze (self .length_scale ))
1207
+ self .__class__ .__name__ , np . ravel (self .length_scale )[ 0 ] )
1217
1208
1218
1209
1219
1210
class Matern (RBF ):
@@ -1359,7 +1350,8 @@ def __repr__(self):
1359
1350
self .nu )
1360
1351
else :
1361
1352
return "{0}(length_scale={1:.3g}, nu={2:.3g})" .format (
1362
- self .__class__ .__name__ , _squeeze (self .length_scale ), self .nu )
1353
+ self .__class__ .__name__ , np .ravel (self .length_scale )[0 ],
1354
+ self .nu )
1363
1355
1364
1356
1365
1357
class RationalQuadratic (StationaryKernelMixin , NormalizedKernelMixin , Kernel ):
You can’t perform that action at this time.
0 commit comments