| op { |
| name: "SdcaOptimizer" |
| input_arg { |
| name: "sparse_example_indices" |
| type: DT_INT64 |
| number_attr: "num_sparse_features" |
| } |
| input_arg { |
| name: "sparse_feature_indices" |
| type: DT_INT64 |
| number_attr: "num_sparse_features" |
| } |
| input_arg { |
| name: "sparse_feature_values" |
| type: DT_FLOAT |
| number_attr: "num_sparse_features_with_values" |
| } |
| input_arg { |
| name: "dense_features" |
| type: DT_FLOAT |
| number_attr: "num_dense_features" |
| } |
| input_arg { |
| name: "example_weights" |
| type: DT_FLOAT |
| } |
| input_arg { |
| name: "example_labels" |
| type: DT_FLOAT |
| } |
| input_arg { |
| name: "sparse_indices" |
| type: DT_INT64 |
| number_attr: "num_sparse_features" |
| } |
| input_arg { |
| name: "sparse_weights" |
| type: DT_FLOAT |
| number_attr: "num_sparse_features" |
| } |
| input_arg { |
| name: "dense_weights" |
| type: DT_FLOAT |
| number_attr: "num_dense_features" |
| } |
| input_arg { |
| name: "example_state_data" |
| type: DT_FLOAT |
| } |
| output_arg { |
| name: "out_example_state_data" |
| type: DT_FLOAT |
| } |
| output_arg { |
| name: "out_delta_sparse_weights" |
| type: DT_FLOAT |
| number_attr: "num_sparse_features" |
| } |
| output_arg { |
| name: "out_delta_dense_weights" |
| type: DT_FLOAT |
| number_attr: "num_dense_features" |
| } |
| attr { |
| name: "loss_type" |
| type: "string" |
| allowed_values { |
| list { |
| s: "logistic_loss" |
| s: "squared_loss" |
| s: "hinge_loss" |
| s: "smooth_hinge_loss" |
| } |
| } |
| } |
| attr { |
| name: "adaptative" |
| type: "bool" |
| default_value { |
| b: false |
| } |
| } |
| attr { |
| name: "num_sparse_features" |
| type: "int" |
| has_minimum: true |
| } |
| attr { |
| name: "num_sparse_features_with_values" |
| type: "int" |
| has_minimum: true |
| } |
| attr { |
| name: "num_dense_features" |
| type: "int" |
| has_minimum: true |
| } |
| attr { |
| name: "l1" |
| type: "float" |
| } |
| attr { |
| name: "l2" |
| type: "float" |
| } |
| attr { |
| name: "num_loss_partitions" |
| type: "int" |
| has_minimum: true |
| minimum: 1 |
| } |
| attr { |
| name: "num_inner_iterations" |
| type: "int" |
| has_minimum: true |
| minimum: 1 |
| } |
| } |
| op { |
| name: "SdcaOptimizer" |
| input_arg { |
| name: "sparse_example_indices" |
| type: DT_INT64 |
| number_attr: "num_sparse_features" |
| } |
| input_arg { |
| name: "sparse_feature_indices" |
| type: DT_INT64 |
| number_attr: "num_sparse_features" |
| } |
| input_arg { |
| name: "sparse_feature_values" |
| type: DT_FLOAT |
| number_attr: "num_sparse_features_with_values" |
| } |
| input_arg { |
| name: "dense_features" |
| type: DT_FLOAT |
| number_attr: "num_dense_features" |
| } |
| input_arg { |
| name: "example_weights" |
| type: DT_FLOAT |
| } |
| input_arg { |
| name: "example_labels" |
| type: DT_FLOAT |
| } |
| input_arg { |
| name: "sparse_indices" |
| type: DT_INT64 |
| number_attr: "num_sparse_features" |
| } |
| input_arg { |
| name: "sparse_weights" |
| type: DT_FLOAT |
| number_attr: "num_sparse_features" |
| } |
| input_arg { |
| name: "dense_weights" |
| type: DT_FLOAT |
| number_attr: "num_dense_features" |
| } |
| input_arg { |
| name: "example_state_data" |
| type: DT_FLOAT |
| } |
| output_arg { |
| name: "out_example_state_data" |
| type: DT_FLOAT |
| } |
| output_arg { |
| name: "out_delta_sparse_weights" |
| type: DT_FLOAT |
| number_attr: "num_sparse_features" |
| } |
| output_arg { |
| name: "out_delta_dense_weights" |
| type: DT_FLOAT |
| number_attr: "num_dense_features" |
| } |
| attr { |
| name: "loss_type" |
| type: "string" |
| allowed_values { |
| list { |
| s: "logistic_loss" |
| s: "squared_loss" |
| s: "hinge_loss" |
| s: "smooth_hinge_loss" |
| s: "poisson_loss" |
| } |
| } |
| } |
| attr { |
| name: "adaptative" |
| type: "bool" |
| default_value { |
| b: false |
| } |
| } |
| attr { |
| name: "num_sparse_features" |
| type: "int" |
| has_minimum: true |
| } |
| attr { |
| name: "num_sparse_features_with_values" |
| type: "int" |
| has_minimum: true |
| } |
| attr { |
| name: "num_dense_features" |
| type: "int" |
| has_minimum: true |
| } |
| attr { |
| name: "l1" |
| type: "float" |
| } |
| attr { |
| name: "l2" |
| type: "float" |
| } |
| attr { |
| name: "num_loss_partitions" |
| type: "int" |
| has_minimum: true |
| minimum: 1 |
| } |
| attr { |
| name: "num_inner_iterations" |
| type: "int" |
| has_minimum: true |
| minimum: 1 |
| } |
| } |