python中shuffleSplit()函数

参数:

n : int

数据集中的元素总数。

n_iter : int (default 10)

重新洗牌和分裂迭代次数。

test_size : float (default 0.1), int, or None

如果是float类型的数据, 这个数应该介于0-1.0之间,代表test集所占比例. 如果是int类型, 代表test集的数量. 如果为None, 值将自动设置为train集大小的补集

train_size : float, int, or None (default is None)

如果是float类型的数据 应该介于0和1之间,并表示数据集在train集分割中所占的比例 如果是int类型, 代表train集的样本数量. 如果为None, 值将自动设置为test集大小的补集

random_state : int or RandomState

用于随机抽样的伪随机数发生器状态。

[python]  view plain  copy
 
  1. >>> from sklearn import cross_validation  
  2. >>> rs = cross_validation.ShuffleSplit(4, n_iter=3,  
  3. ...     test_size=.25, random_state=0)  
  4. >>> len(rs)  
  5. 3  
  6. >>> print(rs)  
  7. ...   
  8. ShuffleSplit(4, n_iter=3, test_size=0.25, ...)  
  9. >>> for train_index, test_index in rs:  
  10. ...    print("TRAIN:", train_index, "TEST:", test_index)  
  11. ...  
  12. TRAIN: [0] TEST: [2]  
  13. TRAIN: [3] TEST: [0]  
  14. TRAIN: [1] TEST: [3]  
 

[python]  view plain  copy
 
  1. >>> rs = cross_validation.ShuffleSplit(4, n_iter=3,  
  2. ...     train_size=0.5, test_size=.25, random_state=0)  
  3. >>> for train_index, test_index in rs:  
  4. ...    print("TRAIN:", train_index, "TEST:", test_index)  
  5. ...  
  6. TRAIN: [1] TEST: [2]  
  7. TRAIN: [1] TEST: [0]  
  8. TRAIN: [2] TEST: [3]  
  9. .. automethod:: __init__  

猜你喜欢

转载自www.cnblogs.com/fujian-code/p/8999305.html