Commit 3c53deb3 authored by Aneesh Pappu's avatar Aneesh Pappu
Browse files

stuff

parent f9f55ca3
Loading
Loading
Loading
Loading
+1 −601

File changed.

Preview size limit exceeded, changes collapsed.

+2 −1
Original line number Diff line number Diff line
@@ -14,6 +14,7 @@ from deepchem.splits import ScaffoldSplitter
from deepchem.splits import StratifiedSplitter
from deepchem.datasets.tests import TestDatasetAPI
import pandas as pd
import math


class TestSplitters(TestDatasetAPI):
@@ -91,7 +92,7 @@ class TestSplitters(TestDatasetAPI):
        sparse_np_list = sparse_dataset.to_numpy()
        sparse_np = sparse_np_list[1]
        frac_train = 0.5
        cutoff = frac_train * len(sparse_np)
        cutoff = int(math.floor(frac_train * len(sparse_np)))
        sparse_np = sparse_np[:cutoff, :]
        sparse_df = pd.DataFrame(data = sparse_np)
        total_rows = len(sparse_df.index)