blob: 3746f9504ac61c85d41f43de1d5fa7cf2bfbe33f [file] [log] [blame]
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
}
}