Commit 67aaa576 authored by leswing's avatar leswing
Browse files

Merge branch 'xgboost-again' into tg-atomconv-cr-out

parents 8bba2baa bd94ae24
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -35,7 +35,7 @@ class TestTensorGraph(unittest.TestCase):
    tg.set_loss(loss)
    tg.fit(dataset, nb_epoch=10)
    prediction = np.squeeze(tg.predict_proba_on_batch(X))
    assert_true(np.all(np.isclose(prediction, y, atol=0.2)))
    assert_true(np.all(np.isclose(prediction, y, atol=0.4)))

  def test_multi_task_classifier(self):
    n_data_points = 20
+1 −1
Original line number Diff line number Diff line
@@ -1010,7 +1010,7 @@ class TestOverfit(test_util.TensorFlowTestCase):
    # Eval model on train
    scores = model.evaluate(dataset, [regression_metric])

    assert scores[regression_metric.name] > .9
    assert scores[regression_metric.name] > .8

  def test_siamese_singletask_classification_overfit(self):
    """Test siamese singletask model overfits tiny data."""
+9 −9
Original line number Diff line number Diff line
@@ -12,7 +12,7 @@ from __future__ import unicode_literals
import numpy as np
import tensorflow as tf
import deepchem
import xgboost
from deepchem.utils.dependencies import xgboost
from deepchem.molnet.preset_hyper_parameters import hps
from sklearn.ensemble import RandomForestClassifier
from sklearn.ensemble import RandomForestRegressor