-
Notifications
You must be signed in to change notification settings - Fork 80
/
model.go
342 lines (265 loc) · 13.1 KB
/
model.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
// Copyright (c) 2016, 2018, 2024, Oracle and/or its affiliates. All rights reserved.
// This software is dual-licensed to you under the Universal Permissive License (UPL) 1.0 as shown at https://oss.oracle.com/licenses/upl or Apache License 2.0 as shown at http://www.apache.org/licenses/LICENSE-2.0. You may choose either license.
// Code generated. DO NOT EDIT.
// Vision API
//
// Using Vision, you can upload images to detect and classify objects in them. If you have lots of images, you can process them in batch using asynchronous API endpoints. Vision's features are thematically split between Document AI for document-centric images, and Image Analysis for object and scene-based images. Pretrained models and custom models are supported.
//
package aivision
import (
"encoding/json"
"fmt"
"github.com/oracle/oci-go-sdk/v65/common"
"strings"
)
// Model Machine-learned Model.
type Model struct {
// A unique identifier that is immutable after creation.
Id *string `mandatory:"true" json:"id"`
// The compartment identifier.
CompartmentId *string `mandatory:"true" json:"compartmentId"`
// What type of Vision model this is.
ModelType ModelModelTypeEnum `mandatory:"true" json:"modelType"`
TrainingDataset Dataset `mandatory:"true" json:"trainingDataset"`
// The version of the model.
ModelVersion *string `mandatory:"true" json:"modelVersion"`
// The OCID (https://docs.cloud.oracle.com/iaas/Content/General/Concepts/identifiers.htm) of the project that contains the model.
ProjectId *string `mandatory:"true" json:"projectId"`
// When the model was created, as an RFC3339 datetime string.
TimeCreated *common.SDKTime `mandatory:"true" json:"timeCreated"`
// The current state of the model.
LifecycleState ModelLifecycleStateEnum `mandatory:"true" json:"lifecycleState"`
// A human-friendly name for the model, which can be changed.
DisplayName *string `mandatory:"false" json:"displayName"`
// An optional description of the model.
Description *string `mandatory:"false" json:"description"`
// Set to true when experimenting with a new model type or dataset, so model training is quick, with a predefined low number of passes through the training data.
IsQuickMode *bool `mandatory:"false" json:"isQuickMode"`
// The maximum model training duration in hours, expressed as a decimal fraction.
MaxTrainingDurationInHours *float64 `mandatory:"false" json:"maxTrainingDurationInHours"`
// The total hours actually used for model training.
TrainedDurationInHours *float64 `mandatory:"false" json:"trainedDurationInHours"`
TestingDataset Dataset `mandatory:"false" json:"testingDataset"`
ValidationDataset Dataset `mandatory:"false" json:"validationDataset"`
// When the model was updated, as an RFC3339 datetime string.
TimeUpdated *common.SDKTime `mandatory:"false" json:"timeUpdated"`
// A message describing the current state in more detail, that can provide actionable information if training failed.
LifecycleDetails *string `mandatory:"false" json:"lifecycleDetails"`
// The precision of the trained model.
Precision *float32 `mandatory:"false" json:"precision"`
// Recall of the trained model.
Recall *float32 `mandatory:"false" json:"recall"`
// The mean average precision of the trained model.
AveragePrecision *float32 `mandatory:"false" json:"averagePrecision"`
// The intersection over the union threshold used for calculating precision and recall.
ConfidenceThreshold *float32 `mandatory:"false" json:"confidenceThreshold"`
// The number of images in the dataset used to train, validate, and test the model.
TotalImageCount *int `mandatory:"false" json:"totalImageCount"`
// The number of images set aside for evaluating model performance metrics after training.
TestImageCount *int `mandatory:"false" json:"testImageCount"`
// The complete set of per-label metrics for successfully trained models.
Metrics *string `mandatory:"false" json:"metrics"`
// A simple key-value pair that is applied without any predefined name, type, or scope. It exists for cross-compatibility only.
// For example: `{"bar-key": "value"}`
FreeformTags map[string]string `mandatory:"false" json:"freeformTags"`
// Defined tags for this resource. Each key is predefined and scoped to a namespace.
// For example: `{"foo-namespace": {"bar-key": "value"}}`
DefinedTags map[string]map[string]interface{} `mandatory:"false" json:"definedTags"`
// Usage of system tag keys. These predefined keys are scoped to namespaces.
// For example: `{"orcl-cloud": {"free-tier-retained": "true"}}`
SystemTags map[string]map[string]interface{} `mandatory:"false" json:"systemTags"`
}
func (m Model) String() string {
return common.PointerString(m)
}
// ValidateEnumValue returns an error when providing an unsupported enum value
// This function is being called during constructing API request process
// Not recommended for calling this function directly
func (m Model) ValidateEnumValue() (bool, error) {
errMessage := []string{}
if _, ok := GetMappingModelModelTypeEnum(string(m.ModelType)); !ok && m.ModelType != "" {
errMessage = append(errMessage, fmt.Sprintf("unsupported enum value for ModelType: %s. Supported values are: %s.", m.ModelType, strings.Join(GetModelModelTypeEnumStringValues(), ",")))
}
if _, ok := GetMappingModelLifecycleStateEnum(string(m.LifecycleState)); !ok && m.LifecycleState != "" {
errMessage = append(errMessage, fmt.Sprintf("unsupported enum value for LifecycleState: %s. Supported values are: %s.", m.LifecycleState, strings.Join(GetModelLifecycleStateEnumStringValues(), ",")))
}
if len(errMessage) > 0 {
return true, fmt.Errorf(strings.Join(errMessage, "\n"))
}
return false, nil
}
// UnmarshalJSON unmarshals from json
func (m *Model) UnmarshalJSON(data []byte) (e error) {
model := struct {
DisplayName *string `json:"displayName"`
Description *string `json:"description"`
IsQuickMode *bool `json:"isQuickMode"`
MaxTrainingDurationInHours *float64 `json:"maxTrainingDurationInHours"`
TrainedDurationInHours *float64 `json:"trainedDurationInHours"`
TestingDataset dataset `json:"testingDataset"`
ValidationDataset dataset `json:"validationDataset"`
TimeUpdated *common.SDKTime `json:"timeUpdated"`
LifecycleDetails *string `json:"lifecycleDetails"`
Precision *float32 `json:"precision"`
Recall *float32 `json:"recall"`
AveragePrecision *float32 `json:"averagePrecision"`
ConfidenceThreshold *float32 `json:"confidenceThreshold"`
TotalImageCount *int `json:"totalImageCount"`
TestImageCount *int `json:"testImageCount"`
Metrics *string `json:"metrics"`
FreeformTags map[string]string `json:"freeformTags"`
DefinedTags map[string]map[string]interface{} `json:"definedTags"`
SystemTags map[string]map[string]interface{} `json:"systemTags"`
Id *string `json:"id"`
CompartmentId *string `json:"compartmentId"`
ModelType ModelModelTypeEnum `json:"modelType"`
TrainingDataset dataset `json:"trainingDataset"`
ModelVersion *string `json:"modelVersion"`
ProjectId *string `json:"projectId"`
TimeCreated *common.SDKTime `json:"timeCreated"`
LifecycleState ModelLifecycleStateEnum `json:"lifecycleState"`
}{}
e = json.Unmarshal(data, &model)
if e != nil {
return
}
var nn interface{}
m.DisplayName = model.DisplayName
m.Description = model.Description
m.IsQuickMode = model.IsQuickMode
m.MaxTrainingDurationInHours = model.MaxTrainingDurationInHours
m.TrainedDurationInHours = model.TrainedDurationInHours
nn, e = model.TestingDataset.UnmarshalPolymorphicJSON(model.TestingDataset.JsonData)
if e != nil {
return
}
if nn != nil {
m.TestingDataset = nn.(Dataset)
} else {
m.TestingDataset = nil
}
nn, e = model.ValidationDataset.UnmarshalPolymorphicJSON(model.ValidationDataset.JsonData)
if e != nil {
return
}
if nn != nil {
m.ValidationDataset = nn.(Dataset)
} else {
m.ValidationDataset = nil
}
m.TimeUpdated = model.TimeUpdated
m.LifecycleDetails = model.LifecycleDetails
m.Precision = model.Precision
m.Recall = model.Recall
m.AveragePrecision = model.AveragePrecision
m.ConfidenceThreshold = model.ConfidenceThreshold
m.TotalImageCount = model.TotalImageCount
m.TestImageCount = model.TestImageCount
m.Metrics = model.Metrics
m.FreeformTags = model.FreeformTags
m.DefinedTags = model.DefinedTags
m.SystemTags = model.SystemTags
m.Id = model.Id
m.CompartmentId = model.CompartmentId
m.ModelType = model.ModelType
nn, e = model.TrainingDataset.UnmarshalPolymorphicJSON(model.TrainingDataset.JsonData)
if e != nil {
return
}
if nn != nil {
m.TrainingDataset = nn.(Dataset)
} else {
m.TrainingDataset = nil
}
m.ModelVersion = model.ModelVersion
m.ProjectId = model.ProjectId
m.TimeCreated = model.TimeCreated
m.LifecycleState = model.LifecycleState
return
}
// ModelModelTypeEnum Enum with underlying type: string
type ModelModelTypeEnum string
// Set of constants representing the allowable values for ModelModelTypeEnum
const (
ModelModelTypeImageClassification ModelModelTypeEnum = "IMAGE_CLASSIFICATION"
ModelModelTypeObjectDetection ModelModelTypeEnum = "OBJECT_DETECTION"
)
var mappingModelModelTypeEnum = map[string]ModelModelTypeEnum{
"IMAGE_CLASSIFICATION": ModelModelTypeImageClassification,
"OBJECT_DETECTION": ModelModelTypeObjectDetection,
}
var mappingModelModelTypeEnumLowerCase = map[string]ModelModelTypeEnum{
"image_classification": ModelModelTypeImageClassification,
"object_detection": ModelModelTypeObjectDetection,
}
// GetModelModelTypeEnumValues Enumerates the set of values for ModelModelTypeEnum
func GetModelModelTypeEnumValues() []ModelModelTypeEnum {
values := make([]ModelModelTypeEnum, 0)
for _, v := range mappingModelModelTypeEnum {
values = append(values, v)
}
return values
}
// GetModelModelTypeEnumStringValues Enumerates the set of values in String for ModelModelTypeEnum
func GetModelModelTypeEnumStringValues() []string {
return []string{
"IMAGE_CLASSIFICATION",
"OBJECT_DETECTION",
}
}
// GetMappingModelModelTypeEnum performs case Insensitive comparison on enum value and return the desired enum
func GetMappingModelModelTypeEnum(val string) (ModelModelTypeEnum, bool) {
enum, ok := mappingModelModelTypeEnumLowerCase[strings.ToLower(val)]
return enum, ok
}
// ModelLifecycleStateEnum Enum with underlying type: string
type ModelLifecycleStateEnum string
// Set of constants representing the allowable values for ModelLifecycleStateEnum
const (
ModelLifecycleStateCreating ModelLifecycleStateEnum = "CREATING"
ModelLifecycleStateUpdating ModelLifecycleStateEnum = "UPDATING"
ModelLifecycleStateActive ModelLifecycleStateEnum = "ACTIVE"
ModelLifecycleStateDeleting ModelLifecycleStateEnum = "DELETING"
ModelLifecycleStateDeleted ModelLifecycleStateEnum = "DELETED"
ModelLifecycleStateFailed ModelLifecycleStateEnum = "FAILED"
)
var mappingModelLifecycleStateEnum = map[string]ModelLifecycleStateEnum{
"CREATING": ModelLifecycleStateCreating,
"UPDATING": ModelLifecycleStateUpdating,
"ACTIVE": ModelLifecycleStateActive,
"DELETING": ModelLifecycleStateDeleting,
"DELETED": ModelLifecycleStateDeleted,
"FAILED": ModelLifecycleStateFailed,
}
var mappingModelLifecycleStateEnumLowerCase = map[string]ModelLifecycleStateEnum{
"creating": ModelLifecycleStateCreating,
"updating": ModelLifecycleStateUpdating,
"active": ModelLifecycleStateActive,
"deleting": ModelLifecycleStateDeleting,
"deleted": ModelLifecycleStateDeleted,
"failed": ModelLifecycleStateFailed,
}
// GetModelLifecycleStateEnumValues Enumerates the set of values for ModelLifecycleStateEnum
func GetModelLifecycleStateEnumValues() []ModelLifecycleStateEnum {
values := make([]ModelLifecycleStateEnum, 0)
for _, v := range mappingModelLifecycleStateEnum {
values = append(values, v)
}
return values
}
// GetModelLifecycleStateEnumStringValues Enumerates the set of values in String for ModelLifecycleStateEnum
func GetModelLifecycleStateEnumStringValues() []string {
return []string{
"CREATING",
"UPDATING",
"ACTIVE",
"DELETING",
"DELETED",
"FAILED",
}
}
// GetMappingModelLifecycleStateEnum performs case Insensitive comparison on enum value and return the desired enum
func GetMappingModelLifecycleStateEnum(val string) (ModelLifecycleStateEnum, bool) {
enum, ok := mappingModelLifecycleStateEnumLowerCase[strings.ToLower(val)]
return enum, ok
}