Skip to content

Commit 45b4478

Browse files
authored
Update config.yaml
1 parent 0f9cf0f commit 45b4478

File tree

1 file changed

+7
-14
lines changed

1 file changed

+7
-14
lines changed

config.yaml

+7-14
Original file line numberDiff line numberDiff line change
@@ -1,31 +1,24 @@
1-
experiment_name: "knn_150_20_65"
1+
experiment_name: "Webservice"
22

33
mini_batch:
4-
size: 4
4+
size: 1
55
# N means the number of cluster text in one inference
66
# strongly recommend to set N to half ~ two-thirds of your VRAM(GB)
77

8-
data:
9-
source: "youtube" # "opensource"
10-
opensource: "ccdv/govreport-summarization"
11-
youtube: "WhiteboardLLM/Data"
12-
index_set: 3
13-
148
segment:
159
args:
16-
n_word: 150
10+
n_word: 30
1711
n_overlap: 0
1812
fix_size: False
1913

2014
concat:
21-
method: "concate_knn"
15+
method: "concate_kmeans"
2216
args:
23-
k: 20
24-
threshold: 0.65
17+
k: 5
2518

2619
summary:
2720
args:
28-
min_length: 100
21+
min_length: 30
2922
max_length: 1024
3023

31-
save_summaries: True
24+
save_summaries: False

0 commit comments

Comments
 (0)