Commit ba3d839b authored by Aneesh Pappu's avatar Aneesh Pappu
Browse files

sparse

parent 38649325

deepchem/splits/tests/test_splitter.py

+1 −0
Original line number Diff line number Diff line
@@ -91,6 +91,7 @@ class TestSplitters(TestDatasetAPI): 
        sparse_dataset = self.load_sparse_multitask_dataset()

        sparse_np_list = sparse_dataset.to_numpy()

        sparse_np = sparse_np_list[1]

        print(sparse_np[:, 3])

        frac_train = 0.5

        cutoff = int(math.floor(frac_train * len(sparse_np)))

        sparse_np = sparse_np[:cutoff, :]

CRA Git | Maintained and supported by SUSTech CRA and CCSE