-
Notifications
You must be signed in to change notification settings - Fork 250
/
experiment.yaml
44 lines (41 loc) · 1.28 KB
/
experiment.yaml
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
name: web_classification
flow: flows/experiment
connections:
- name: aoai
connection_type: AzureOpenAIConnection
api_base: https://demoopenaiexamples.openai.azure.com/
api_version: 2023-07-01-preview
api_key: ${api_key}
api_type: azure
datasets:
- name: web_classification_train
source: data/data.jsonl
description: "This dataset is for prompt experiments."
mappings:
url: "${data.url}"
- name: web_classification_train_new
source: data/data_new.jsonl
description: "This dataset is for prompt experiments."
mappings:
url: "${data.url}"
evaluators:
- name: web_classification_flow
flow: flows/evaluation
datasets:
- name: web_classification_train_eval
reference: web_classification_train
source: data/eval_data.jsonl
description: "This dataset is for evaluating flows."
mappings:
groundtruth: "${data.answer}"
prediction: "${run.outputs.category}"
- name: web_classification_flow_adv
flow: flows/evaluation_adv
datasets:
- name: web_classification_train_new_eval
reference: web_classification_train_new
source: data/eval_data_new.jsonl
description: "This dataset is for evaluating flows."
mappings:
groundtruth: "${data.answer}"
prediction: "${run.outputs.category}"