def __init__(self, fm_decoder, n_iter=5, test_size=0.2, train_size=None,
random_state=None):
self.fm_decoder = fm_decoder
StratifiedShuffleSplit.__init__(
self,
n_iter=n_iter,
test_size=test_size,
train_size=train_size,
random_state=random_state)
