24
24
from ..utils import indexable , check_random_state , safe_indexing
25
25
from ..utils .validation import _num_samples , column_or_1d
26
26
from ..utils .validation import check_array
27
- from ..utils .validation import SCALAR_INTEGER_TYPES , floating_types
27
+ from ..utils .validation import SCALAR_INTEGER_TYPES , SCALAR_FLOATING_TYPES
28
28
from ..utils .multiclass import type_of_target
29
29
from ..externals .six import with_metaclass
30
30
from ..externals .six .moves import zip
@@ -1643,7 +1643,7 @@ def _validate_shuffle_split_init(test_size, train_size):
1643
1643
raise ValueError ('test_size and train_size can not both be None' )
1644
1644
1645
1645
if test_size is not None :
1646
- if isinstance (test_size , floating_types ):
1646
+ if isinstance (test_size , SCALAR_FLOATING_TYPES ):
1647
1647
if test_size >= 1. :
1648
1648
raise ValueError (
1649
1649
'test_size=%f should be smaller '
@@ -1653,11 +1653,11 @@ def _validate_shuffle_split_init(test_size, train_size):
1653
1653
raise ValueError ("Invalid value for test_size: %r" % test_size )
1654
1654
1655
1655
if train_size is not None :
1656
- if isinstance (train_size , floating_types ):
1656
+ if isinstance (train_size , SCALAR_FLOATING_TYPES ):
1657
1657
if train_size >= 1. :
1658
1658
raise ValueError ("train_size=%f should be smaller "
1659
1659
"than 1.0 or be an integer" % train_size )
1660
- elif (isinstance (test_size , floating_types ) and
1660
+ elif (isinstance (test_size , SCALAR_FLOATING_TYPES ) and
1661
1661
(train_size + test_size ) > 1. ):
1662
1662
raise ValueError ('The sum of test_size and train_size = %f, '
1663
1663
'should be smaller than 1.0. Reduce '
@@ -1690,14 +1690,14 @@ def _validate_shuffle_split(n_samples, test_size, train_size):
1690
1690
if test_size == "default" :
1691
1691
test_size = 0.1
1692
1692
1693
- if isinstance (test_size , floating_types ):
1693
+ if isinstance (test_size , SCALAR_FLOATING_TYPES ):
1694
1694
n_test = ceil (test_size * n_samples )
1695
1695
elif isinstance (test_size , SCALAR_INTEGER_TYPES ):
1696
1696
n_test = float (test_size )
1697
1697
1698
1698
if train_size is None :
1699
1699
n_train = n_samples - n_test
1700
- elif isinstance (train_size , floating_types ):
1700
+ elif isinstance (train_size , SCALAR_FLOATING_TYPES ):
1701
1701
n_train = floor (train_size * n_samples )
1702
1702
else :
1703
1703
n_train = float (train_size )
0 commit comments