Commit c23df2b4 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by Bharath Ramsundar
Browse files

Putting together some reload tests

parent ae12a7eb
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -344,8 +344,8 @@ class MultitaskRegressor(KerasModel):
class MultitaskFitTransformRegressor(MultitaskRegressor):
  """Implements a MultitaskRegressor that performs on-the-fly transformation during fit/predict.

  Example:

  Examples
  --------
  >>> n_samples = 10
  >>> n_features = 3
  >>> n_tasks = 1
+3 −3
Original line number Diff line number Diff line
@@ -266,7 +266,7 @@ def test_skewed_classification_overfit():


def test_skewed_missing_classification_overfit():
  """TG, skewed data, few actives
  """MultitaskClassifier, skewed data, few actives

  Test MultitaskClassifier overfit 0/1 datasets with missing data and few
  actives. This is intended to be as close to singletask MUV datasets as
@@ -377,8 +377,8 @@ def test_multitask_classification_overfit():
  assert scores[classification_metric.name] > .9


def test_tf_robust_multitask_classification_overfit():
  """Test tf robust multitask overfits tiny data."""
def test_robust_multitask_classification_overfit():
  """Test robust multitask overfits tiny data."""
  n_tasks = 10
  n_samples = 10
  n_features = 3
+747 −30

File changed.

Preview size limit exceeded, changes collapsed.