-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathtrained_model_aliases.yml
43 lines (43 loc) · 1.29 KB
/
trained_model_aliases.yml
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
---
requires:
serverless: true
stack: true
---
teardown:
- do:
ml.delete_trained_model: { model_id: a-classification-model, force: true }
---
'model aliases':
- do:
ml.put_trained_model:
model_id: a-classification-model
body: >
{
"description": "empty model for tests",
"input": {"field_names": ["field1", "field2"]},
"tags": ["classification", "tag2"],
"inference_config": {"classification": {}},
"definition": {
"preprocessors": [],
"trained_model": {
"tree": {
"feature_names": ["field1", "field2"],
"tree_structure": [
{"node_index": 0, "leaf_value": 1}
],
"target_type": "classification",
"classification_labels": ["no", "yes"]
}
}
}
}
- do:
ml.put_trained_model_alias:
model_alias: "alias-to-a-classification-model"
model_id: "a-classification-model"
- is_true: 'acknowledged'
- do:
ml.delete_trained_model_alias:
model_alias: "alias-to-a-classification-model"
model_id: "a-classification-model"
- is_true: 'acknowledged'