Commit 4f14aebc authored by miaecle's avatar miaecle
Browse files

little modification

parent 631dbe69
Loading
Loading
Loading
Loading
+1 −2
Original line number Diff line number Diff line
@@ -329,7 +329,7 @@ class IndexSplitter(Splitter):

class IndiceSplitter(Splitter):
  """
  Class for simple order based splits. 
  Class for splits based on input order. 
  """
  def __init__(self, verbose=False, valid_indices=None, test_indices=None):
    """
@@ -348,7 +348,6 @@ class IndiceSplitter(Splitter):
            frac_test=.1, log_every_n=None):
    """
    Splits internal compounds into train/validation/test in designated order.
        
    """
    num_datapoints = len(dataset)
    indices = np.arange(num_datapoints).tolist()