Skip to content

Commit 4ff353c

Browse files
committed
fix?
1 parent 723c6ae commit 4ff353c

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

‎ostap/stats/adval.py‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -811,7 +811,6 @@ def __init__ ( self ,
811811
'scoring' : 'neg_root_mean_squared_error',
812812
}
813813

814-
815814
config.update ( params )
816815

817816
import sklearn
@@ -1025,7 +1024,12 @@ def __init__ ( self ,
10251024
'bootstrap' : True ,
10261025
'max_samples' : 0.8 ,
10271026
}
1028-
config.update ( params )
1027+
config.update ( params )
1028+
1029+
import sklearn
1030+
from packaging.version import Version
1031+
if Version ( sklearn.__version__ ) < Version ( '1.0' ) :
1032+
if 'squared_error' == config.get ( 'loss' ) : config [ 'loss' ] = 'mse'
10291033

10301034
ADVAL_base.__init__ ( self,
10311035
nToys = nToys ,

0 commit comments

Comments
 (0)