Skip to content

Commit ff1133a

Browse files
committed
Add recipes/RTX_3060_12GB/sentiment-analyzer-multilingual.yaml
1 parent 934eec1 commit ff1133a

File tree

1 file changed

+37
-0
lines changed

1 file changed

+37
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
target_task: tasks/nlp/sentiment-analysis.md
2+
base_model_id: TinyLlama/TinyLlama-1.1B-intermediate-step-715k-1.5T
3+
model_name: tinyllama-sentiment-analyzer-multilingual-1.5T-v0.1
4+
output_base_dir: output
5+
dataset_id: tyqiangz/multilingual-sentiments
6+
dataset_load_config: all
7+
dataset_input_field_name: text
8+
dataset_output_field_name: label
9+
dataset_output_field_values_to_texts:
10+
0: "Positive"
11+
1: "Neutral"
12+
2: "Negative"
13+
dataset_train_split_seed: 42
14+
dataset_train_split_test_size: 0.2
15+
lora_r: 8
16+
lora_alpha: 16
17+
lora_dropout: 0.05
18+
train_claim_gpu_num: 2
19+
train_per_device_train_batch_size: 6
20+
train_gradient_accumulation_steps: 2
21+
train_num_train_epochs: 4
22+
train_max_steps: 1000
23+
train_fp16: True
24+
inference_max_new_tokens: 2
25+
evaluations:
26+
-
27+
prompt: "Yes my laptop works So now i can abort my diplomthesis"
28+
expected_output: "Negative"
29+
-
30+
prompt: "Hmm don't know what to do today -- Model just canceled = I hate it when that happens..will have to do gardening!"
31+
expected_output: "Negative"
32+
-
33+
prompt: "ぼけっとしてたらこんな時間。チャリあるから食べにでたいのに…"
34+
expected_output: "Negative"
35+
-
36+
prompt: "今日の月も白くて明るい。昨日より雲が少なくてキレイな〜 と立ち止まる帰り道。チャリなし生活も悪くない。"
37+
expected_output: "Positive"

0 commit comments

Comments
 (0)