Commit ff8acdca authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #712 from lilleswing/rl-fatfinger 

initial_state fatfinger
parents a8c39a33 a0ef537a

deepchem/models/tensorgraph/tensor_graph.py

+1 −1
Original line number Diff line number Diff line
@@ -233,7 +233,7 @@ class TensorGraph(Model): 
          feed_dict[self.features[0]] = X_b

        if len(self.task_weights) == 1 and w_b is not None and not predict:

          feed_dict[self.task_weights[0]] = w_b

        for (inital_state, zero_state) in zip(self.rnn_initial_states,

        for (initial_state, zero_state) in zip(self.rnn_initial_states,

                                               self.rnn_zero_states):

          feed_dict[initial_state] = zero_state

        yield feed_dict

deepchem/models/tests/test_overfit.py

+1 −0
Original line number Diff line number Diff line
@@ -696,6 +696,7 @@ class TestOverfit(test_util.TensorFlowTestCase): 
    scores = model.evaluate(dataset, [regression_metric])

    assert scores[regression_metric.name] > .7



  @flaky

  def test_tf_multitask_regression_overfit(self):

    """Test tf multitask overfits tiny data."""

    n_tasks = 10

CRA Git | Maintained and supported by SUSTech CRA and CCSE