// Copyright 2018 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

syntax = "proto3";

package google.cloud.automl.v1beta1;

import "google/api/annotations.proto";
import "google/cloud/automl/v1beta1/classification.proto";
import "google/protobuf/timestamp.proto";

option go_package = "google.golang.org/genproto/googleapis/cloud/automl/v1beta1;automl";
option java_multiple_files = true;
option java_outer_classname = "ImageProto";
option java_package = "com.google.cloud.automl.v1beta1";


// Dataset metadata that is specific to image classification.
message ImageClassificationDatasetMetadata {
  // Required.
  // Type of the classification problem.
  ClassificationType classification_type = 1;
}

// Model metadata for image classification.
message ImageClassificationModelMetadata {
  // Optional. The ID of the `base` model. If it is specified, the new model
  // will be created based on the `base` model. Otherwise, the new model will be
  // created from scratch. The `base` model is expected to be in the same
  // `project` and `location` as the new model to create.
  string base_model_id = 1;

  // Required. The train budget of creating this model. The actual
  // `train_cost` will be equal or less than this value.
  int64 train_budget = 2;

  // Output only. The actual train cost of creating this model. If this
  // model is created from a `base` model, the train cost used to create the
  // `base` model are not included.
  int64 train_cost = 3;

  // Output only. The reason that this create model operation stopped,
  // e.g. BUDGET_REACHED, CONVERGED.
  string stop_reason = 5;
}
