Commit 2e848146 authored by Yutong Zhao's avatar Yutong Zhao
Browse files

Remove print statements

parent ed64a021

deepchem/splits/splitters.py

+0 −2
Original line number Diff line number Diff line
@@ -259,8 +259,6 @@ class RandomGroupSplitter(Splitter): 
    valid_groups = shuffled_group_idxs[train_cutoff:valid_cutoff]

    test_groups = shuffled_group_idxs[valid_cutoff:]



    print(train_groups, valid_groups, test_groups)



    train_idxs = list(itertools.chain(*group_idxs[train_groups]))

    valid_idxs = list(itertools.chain(*group_idxs[valid_groups]))

    test_idxs = list(itertools.chain(*group_idxs[test_groups]))

deepchem/splits/tests/test_splitter.py

+0 −5
Original line number Diff line number Diff line
@@ -36,17 +36,12 @@ class TestSplitters(unittest.TestCase): 
    train_idxs, valid_idxs, test_idxs = group_splitter.split(

        solubility_dataset, frac_train=0.5, frac_valid=0.25, frac_test=0.25)



    print(train_idxs, valid_idxs, test_idxs)

    print(len(train_idxs), len(valid_idxs), len(test_idxs))



    class_ind = [-1] * 10



    all_idxs = []

    for s in train_idxs + valid_idxs + test_idxs:

      all_idxs.append(s)



    print(sorted(all_idxs))



    assert sorted(all_idxs) == list(range(10))



    for split_idx, split in enumerate([train_idxs, valid_idxs, test_idxs]):

CRA Git | Maintained and supported by SUSTech CRA and CCSE