Uncomment @estimator_export decorators related to BaselineEstimator.

API compat test now properly points towards modules in tensorflow_estimator
repo. Uncommenting decorators to get proper API file review.

PiperOrigin-RevId: 217798271
This commit is contained in:
Michael Case 2018-10-18 17:28:45 -07:00 committed by TensorFlower Gardener
parent fda1b4a91f
commit 79bff06f77
4 changed files with 132 additions and 0 deletions

View File

@ -0,0 +1,62 @@
path: "tensorflow.estimator.BaselineEstimator"
tf_class {
is_instance: "<class \'tensorflow_estimator.python.estimator.canned.baseline.BaselineEstimator\'>"
is_instance: "<class \'tensorflow_estimator.python.estimator.estimator.Estimator\'>"
is_instance: "<type \'object\'>"
member {
name: "config"
mtype: "<type \'property\'>"
}
member {
name: "model_dir"
mtype: "<type \'property\'>"
}
member {
name: "model_fn"
mtype: "<type \'property\'>"
}
member {
name: "params"
mtype: "<type \'property\'>"
}
member_method {
name: "__init__"
argspec: "args=[\'self\', \'head\', \'model_dir\', \'optimizer\', \'config\'], varargs=None, keywords=None, defaults=[\'None\', \'Ftrl\', \'None\'], "
}
member_method {
name: "eval_dir"
argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "evaluate"
argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\'], "
}
member_method {
name: "export_saved_model"
argspec: "args=[\'self\', \'export_dir_base\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=None, keywords=None, defaults=[\'None\', \'False\', \'None\'], "
}
member_method {
name: "export_savedmodel"
argspec: "args=[\'self\', \'export_dir_base\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_default_attrs\'], varargs=None, keywords=None, defaults=[\'None\', \'False\', \'None\', \'False\'], "
}
member_method {
name: "get_variable_names"
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "get_variable_value"
argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "latest_checkpoint"
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "predict"
argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_single_examples\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'True\'], "
}
member_method {
name: "train"
argspec: "args=[\'self\', \'input_fn\', \'hooks\', \'steps\', \'max_steps\', \'saving_listeners\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\'], "
}
}

View File

@ -4,6 +4,10 @@ tf_module {
name: "BaselineClassifier"
mtype: "<type \'type\'>"
}
member {
name: "BaselineEstimator"
mtype: "<type \'type\'>"
}
member {
name: "BaselineRegressor"
mtype: "<type \'type\'>"

View File

@ -0,0 +1,62 @@
path: "tensorflow.estimator.BaselineEstimator"
tf_class {
is_instance: "<class \'tensorflow_estimator.python.estimator.canned.baseline.BaselineEstimator\'>"
is_instance: "<class \'tensorflow_estimator.python.estimator.estimator.Estimator\'>"
is_instance: "<type \'object\'>"
member {
name: "config"
mtype: "<type \'property\'>"
}
member {
name: "model_dir"
mtype: "<type \'property\'>"
}
member {
name: "model_fn"
mtype: "<type \'property\'>"
}
member {
name: "params"
mtype: "<type \'property\'>"
}
member_method {
name: "__init__"
argspec: "args=[\'self\', \'head\', \'model_dir\', \'optimizer\', \'config\'], varargs=None, keywords=None, defaults=[\'None\', \'Ftrl\', \'None\'], "
}
member_method {
name: "eval_dir"
argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
}
member_method {
name: "evaluate"
argspec: "args=[\'self\', \'input_fn\', \'steps\', \'hooks\', \'checkpoint_path\', \'name\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\'], "
}
member_method {
name: "export_saved_model"
argspec: "args=[\'self\', \'export_dir_base\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\'], varargs=None, keywords=None, defaults=[\'None\', \'False\', \'None\'], "
}
member_method {
name: "export_savedmodel"
argspec: "args=[\'self\', \'export_dir_base\', \'serving_input_receiver_fn\', \'assets_extra\', \'as_text\', \'checkpoint_path\', \'strip_default_attrs\'], varargs=None, keywords=None, defaults=[\'None\', \'False\', \'None\', \'False\'], "
}
member_method {
name: "get_variable_names"
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "get_variable_value"
argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "latest_checkpoint"
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "predict"
argspec: "args=[\'self\', \'input_fn\', \'predict_keys\', \'hooks\', \'checkpoint_path\', \'yield_single_examples\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'True\'], "
}
member_method {
name: "train"
argspec: "args=[\'self\', \'input_fn\', \'hooks\', \'steps\', \'max_steps\', \'saving_listeners\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\'], "
}
}

View File

@ -4,6 +4,10 @@ tf_module {
name: "BaselineClassifier"
mtype: "<type \'type\'>"
}
member {
name: "BaselineEstimator"
mtype: "<type \'type\'>"
}
member {
name: "BaselineRegressor"
mtype: "<type \'type\'>"