Commit 4019d552 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Fixing scscore

parent 57684218
Loading
Loading
Loading
Loading
+4 −4
Original line number Diff line number Diff line
import unittest

import deepchem
import tempfile
import deepchem as dc
import numpy as np


@@ -16,9 +16,9 @@ class TestScScoreModel(unittest.TestCase):

    X = np.random.rand(n_samples, 2, n_features)
    y = np.random.randint(2, size=(n_samples, n_tasks))
    dataset = deepchem.data.NumpyDataset(X, y)
    dataset = dc.data.NumpyDataset(X, y)

    model = deepchem.models.ScScoreModel(n_features, dropouts=0)
    model = dc.models.ScScoreModel(n_features, dropouts=0)

    model.fit(dataset, nb_epoch=100)
    pred = model.predict(dataset)