File tree 2 files changed +13
-4
lines changed 2 files changed +13
-4
lines changed Original file line number Diff line number Diff line change @@ -1204,7 +1204,7 @@ def __repr__(self):
1204
1204
self .length_scale )))
1205
1205
else : # isotropic
1206
1206
return "{0}(length_scale={1:.3g})" .format (
1207
- self .__class__ .__name__ , self .length_scale )
1207
+ self .__class__ .__name__ , np . ravel ( self .length_scale )[ 0 ] )
1208
1208
1209
1209
1210
1210
class Matern (RBF ):
@@ -1348,9 +1348,10 @@ def __repr__(self):
1348
1348
self .__class__ .__name__ ,
1349
1349
", " .join (map ("{0:.3g}" .format , self .length_scale )),
1350
1350
self .nu )
1351
- else : # isotropic
1351
+ else :
1352
1352
return "{0}(length_scale={1:.3g}, nu={2:.3g})" .format (
1353
- self .__class__ .__name__ , self .length_scale , self .nu )
1353
+ self .__class__ .__name__ , np .ravel (self .length_scale )[0 ],
1354
+ self .nu )
1354
1355
1355
1356
1356
1357
class RationalQuadratic (StationaryKernelMixin , NormalizedKernelMixin , Kernel ):
Original file line number Diff line number Diff line change 41
41
4.0 * Matern (length_scale = [0.5 , 0.5 ], nu = 2.5 ),
42
42
RationalQuadratic (length_scale = 0.5 , alpha = 1.5 ),
43
43
ExpSineSquared (length_scale = 0.5 , periodicity = 1.5 ),
44
- DotProduct (sigma_0 = 2.0 ), DotProduct (sigma_0 = 2.0 ) ** 2 ]
44
+ DotProduct (sigma_0 = 2.0 ), DotProduct (sigma_0 = 2.0 ) ** 2 ,
45
+ RBF (length_scale = [2.0 ]), Matern (length_scale = [2.0 ])]
45
46
for metric in PAIRWISE_KERNEL_FUNCTIONS :
46
47
if metric in ["additive_chi2" , "chi2" ]:
47
48
continue
@@ -304,3 +305,10 @@ def test_set_get_params():
304
305
kernel .set_params (** {hyperparameter .name : value })
305
306
assert_almost_equal (np .exp (kernel .theta [index ]), value )
306
307
index += 1
308
+
309
+
310
+ def test_repr_kernels ():
311
+ """Smoke-test for repr in kernels."""
312
+
313
+ for kernel in kernels :
314
+ repr (kernel )
You can’t perform that action at this time.
0 commit comments