Fixed failing test cases (e0e6b3dd) · Commits · 钟慕尧 / deepchem

deepchem/models/tensorgraph/tests/test_estimators.py

+10 −5

Original line number	Diff line number	Diff line
		@@ -26,7 +26,8 @@ class TestEstimators(unittest.TestCase):
		np.random.seed(123)
		X = np.random.rand(n_samples, n_features)
		y = np.zeros((n_samples, n_tasks))
		dataset = dc.data.NumpyDataset(X, y)
		w = np.ones((n_samples, n_tasks))
		dataset = dc.data.NumpyDataset(X, y, w)

		def input_fn(epochs):
		x, y, weights = dataset.make_iterator(
		@@ -70,7 +71,8 @@ class TestEstimators(unittest.TestCase):
		np.random.seed(123)
		X = np.random.rand(n_samples, n_features)
		y = np.zeros((n_samples, n_tasks))
		dataset = dc.data.NumpyDataset(X, y)
		w = np.ones((n_samples, n_tasks))
		dataset = dc.data.NumpyDataset(X, y, w)

		def input_fn(epochs):
		x, y, weights = dataset.make_iterator(
		@@ -110,7 +112,8 @@ class TestEstimators(unittest.TestCase):
		np.random.seed(123)
		X = np.random.rand(n_samples, n_features)
		y = np.zeros((n_samples, n_tasks))
		dataset = dc.data.NumpyDataset(X, y)
		w = np.ones((n_samples, n_tasks))
		dataset = dc.data.NumpyDataset(X, y, w)

		def input_fn(epochs):
		x, y, weights = dataset.make_iterator(
		@@ -161,7 +164,8 @@ class TestEstimators(unittest.TestCase):
		np.random.seed(123)
		X = np.random.rand(n_samples, n_features)
		y = np.zeros((n_samples, n_tasks))
		dataset = dc.data.NumpyDataset(X, y)
		w = np.ones((n_samples, n_tasks))
		dataset = dc.data.NumpyDataset(X, y, w)

		def input_fn(epochs):
		x, y, weights = dataset.make_iterator(
		@@ -245,7 +249,8 @@ class TestEstimators(unittest.TestCase):
		np.random.seed(123)
		X = np.random.rand(n_samples, n_features)
		y = np.zeros((n_samples, n_tasks))
		dataset = dc.data.NumpyDataset(X, y)
		w = np.ones((n_samples, n_tasks))
		dataset = dc.data.NumpyDataset(X, y, w)
		transformers = [dc.trans.IRVTransformer(10, n_tasks, dataset)]

		for transformer in transformers:

deepchem/models/tensorgraph/tests/test_ontology.py

+4 −2

Original line number	Diff line number	Diff line
		@@ -20,7 +20,8 @@ class TestOntology(test_util.TensorFlowTestCase):
		y[:, 0] = np.sum(x[:, :(n_features // 2)], axis=1)
		y[:, 1] = np.sum(x[:, (n_features // 2):], axis=1)
		y[:, 2] = 0.5 * np.sum(x, axis=1)
		dataset = dc.data.NumpyDataset(x, y)
		w = np.ones((n_samples, n_tasks))
		dataset = dc.data.NumpyDataset(x, y, w)

		# Create an OntologyModel. Two leaf nodes contain half the features.

		@@ -69,7 +70,8 @@ class TestOntology(test_util.TensorFlowTestCase):
		y[:, 0] = np.sum(x[:, :(n_features // 2)], axis=1) < n_features / 4
		y[:, 1] = np.sum(x[:, (n_features // 2):], axis=1) < n_features / 4
		y[:, 2] = np.sum(x, axis=1) < n_features / 2
		dataset = dc.data.NumpyDataset(x, y)
		w = np.ones((n_samples, n_tasks))
		dataset = dc.data.NumpyDataset(x, y, w)

		# Create an OntologyModel. Two leaf nodes contain half the features.

Admin message