--- /dev/null
+// 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";
+option php_namespace = "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;
+}