aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/linear_optimizer
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-03-08 13:36:46 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-03-08 13:41:15 -0800
commitebf554ff77bc46bfdd9b424bc44b62f803100b33 (patch)
treec2374f65d40402815de85d4ec55e1456d047578a /tensorflow/contrib/linear_optimizer
parentd6f3a547af2060974a1397052809a1a7f1e2d311 (diff)
Make adaptive SDCA the default.
PiperOrigin-RevId: 188380039
Diffstat (limited to 'tensorflow/contrib/linear_optimizer')
-rw-r--r--tensorflow/contrib/linear_optimizer/python/kernel_tests/sdca_ops_test.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/contrib/linear_optimizer/python/kernel_tests/sdca_ops_test.py b/tensorflow/contrib/linear_optimizer/python/kernel_tests/sdca_ops_test.py
index 70f777f08b..cfe62fac43 100644
--- a/tensorflow/contrib/linear_optimizer/python/kernel_tests/sdca_ops_test.py
+++ b/tensorflow/contrib/linear_optimizer/python/kernel_tests/sdca_ops_test.py
@@ -270,14 +270,14 @@ class SdcaWithLogisticLossTest(SdcaModelTest):
train_op = lr.minimize()
- def Minimize():
+ def minimize():
with self._single_threaded_test_session():
for _ in range(_MAX_ITERATIONS):
- train_op.run()
+ train_op.run() # pylint: disable=cell-var-from-loop
threads = []
for _ in range(num_loss_partitions):
- threads.append(threading.Thread(target=Minimize))
+ threads.append(threading.Thread(target=minimize))
threads[-1].start()
for t in threads:
@@ -395,7 +395,7 @@ class SdcaWithLogisticLossTest(SdcaModelTest):
predicted_labels = get_binary_predictions_for_logistic(predictions)
self.assertAllClose([0, 1, 1, 1], predicted_labels.eval())
self.assertAllClose(
- 0.01, lr.approximate_duality_gap().eval(), rtol=1e-2, atol=1e-2)
+ 0.0, lr.approximate_duality_gap().eval(), rtol=1e-2, atol=1e-2)
def testFractionalExampleLabel(self):
# Setup test data with 1 positive, and 1 mostly-negative example.
@@ -407,7 +407,7 @@ class SdcaWithLogisticLossTest(SdcaModelTest):
make_example_proto({
'age': [1],
'gender': [1]
- }, 1),
+ }, 0.9),
]
example_weights = [1.0, 1.0]
for num_shards in _SHARD_NUMBERS: