Commit 650843e5 authored by leswing's avatar leswing
Browse files

Also RandomStratifiedSplitter

parent a4f345a6

deepchem/splits/splitters.py

+1 −1
Original line number Diff line number Diff line
@@ -278,7 +278,7 @@ class RandomStratifiedSplitter(Splitter): 


    return train_dataset, valid_dataset, test_dataset



  def k_fold_split(self, dataset, k, directories=None):

  def k_fold_split(self, dataset, k, directories=None, **kwargs):

    """Needs custom implementation due to ragged splits for stratification."""

    log("Computing K-fold split", self.verbose)

    if directories is None:

CRA Git | Maintained and supported by SUSTech CRA and CCSE