experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Migrate the SDCA optimizer to tensorflow_estimator.

Browse Source 
PiperOrigin-RevId: 218718971
This commit is contained in:
A. Unique TensorFlower 2018-10-25 11:18:11 -07:00 committed by TensorFlower Gardener
parent 39b4311ce7
commit 08115e39f4
5 changed files with 38 additions and 1 deletions

5
tensorflow/python/BUILD
View File

@ -1801,7 +1801,10 @@ tf_gen_op_wrapper_private_py(
tf_gen_op_wrapper_private_py(
name = "sdca_ops_gen",
visibility = ["//tensorflow/contrib/linear_optimizer:__pkg__"],
visibility = [
"//tensorflow/contrib/linear_optimizer:__pkg__",
"//tensorflow_estimator/python/estimator/canned/linear_optimizer:__pkg__",
],
)
tf_gen_op_wrapper_private_py(

13
tensorflow/tools/api/golden/v1/tensorflow.estimator.experimental.-linear-s-d-c-a.pbtxt Normal file
View File

@ -0,0 +1,13 @@
path: "tensorflow.estimator.experimental.LinearSDCA"
tf_class {
is_instance: "<class \'tensorflow_estimator.python.estimator.canned.linear.LinearSDCA\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
argspec: "args=[\'self\', \'example_id_column\', \'num_loss_partitions\', \'num_table_shards\', \'symmetric_l1_regularization\', \'symmetric_l2_regularization\', \'adaptive\'], varargs=None, keywords=None, defaults=[\'1\', \'None\', \'0.0\', \'1.0\', \'False\'], "
}
member_method {
name: "get_train_step"
argspec: "args=[\'self\', \'state_manager\', \'weight_column_name\', \'loss_type\', \'feature_columns\', \'features\', \'targets\', \'bias_var\', \'global_step\'], varargs=None, keywords=None, defaults=None"
}
}

4
tensorflow/tools/api/golden/v1/tensorflow.estimator.experimental.pbtxt
View File

@ -1,5 +1,9 @@
path: "tensorflow.estimator.experimental"
tf_module {
member {
name: "LinearSDCA"
mtype: "<type \'type\'>"
}
member_method {
name: "call_logit_fn"
argspec: "args=[\'logit_fn\', \'features\', \'mode\', \'params\', \'config\'], varargs=None, keywords=None, defaults=None"

13
tensorflow/tools/api/golden/v2/tensorflow.estimator.experimental.-linear-s-d-c-a.pbtxt Normal file
View File

@ -0,0 +1,13 @@
path: "tensorflow.estimator.experimental.LinearSDCA"
tf_class {
is_instance: "<class \'tensorflow_estimator.python.estimator.canned.linear.LinearSDCA\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
argspec: "args=[\'self\', \'example_id_column\', \'num_loss_partitions\', \'num_table_shards\', \'symmetric_l1_regularization\', \'symmetric_l2_regularization\', \'adaptive\'], varargs=None, keywords=None, defaults=[\'1\', \'None\', \'0.0\', \'1.0\', \'False\'], "
}
member_method {
name: "get_train_step"
argspec: "args=[\'self\', \'state_manager\', \'weight_column_name\', \'loss_type\', \'feature_columns\', \'features\', \'targets\', \'bias_var\', \'global_step\'], varargs=None, keywords=None, defaults=None"
}
}

4
tensorflow/tools/api/golden/v2/tensorflow.estimator.experimental.pbtxt
View File

@ -1,5 +1,9 @@
path: "tensorflow.estimator.experimental"
tf_module {
member {
name: "LinearSDCA"
mtype: "<type \'type\'>"
}
member_method {
name: "call_logit_fn"
argspec: "args=[\'logit_fn\', \'features\', \'mode\', \'params\', \'config\'], varargs=None, keywords=None, defaults=None"