Commit 7f799c73 authored by Kevin Shen's avatar Kevin Shen
Browse files

added docs and tests

parent dec943af
Loading
Loading
Loading
Loading
+4 −0
Original line number Diff line number Diff line
@@ -50,6 +50,10 @@ class ValidationCallback(object):
      if True, the best model is considered to be the one that minimizes the
      validation metric.  If False, the best model is considered to be the one
      that maximizes it.
    transformers: List[Transformer]
      List of `dc.trans.Transformer` objects. These transformations
      must have been applied to `dataset` previously. The dataset will
      be untransformed for metric evaluation.
    """
    self.dataset = dataset
    self.interval = interval
+2 −2
Original line number Diff line number Diff line
@@ -2,7 +2,6 @@ import unittest
import tempfile
import deepchem as dc
import numpy as np
import tensorflow as tf
try:
  from StringIO import StringIO
except ImportError:
@@ -28,7 +27,8 @@ class TestCallbacks(unittest.TestCase):
        30, [metric],
        log,
        save_dir=save_dir,
        save_on_minimum=False)
        save_on_minimum=False,
        transformers=transformers)
    model.fit(train_dataset, callbacks=callback)

    # Parse the log to pull out the AUC scores.