Views
No views yet
[!NOTE] Detailed articles and the full HES-Ja dataset are scheduled for public release in the near future.
| Model | BLEU | ROUGE-2 | ROUGE-L |
|---|---|---|---|
| Qwen3-0.6B | 0.0147 | 0.0419 | 0.0837 |
| gemma-3-270m-it | 0.0328 | 0.0788 | 0.1549 |
| granite-4.0-350m | 0.0343 | 0.0728 | 0.1575 |
| sarashina2.2-0.5b-instruct-v0.1 (Base) | 0.0263 | 0.0731 | 0.1317 |
| HACHI-Summary-Ja | 0.0276 | 0.0747 | 0.1363 |
| HACHI-Summary-Ja (approx. 100 chars) | 0.0412 | 0.0748 | 0.2055 |
| Model | BLEU | ROUGE-2 | ROUGE-L |
|---|---|---|---|
| Qwen3-0.6B | 0.0963 | 0.1496 | 0.1820 |
| gemma-3-270m-it | 0.1960 | 0.3029 | 0.3403 |
| granite-4.0-350m | 0.1003 | 0.1963 | 0.2330 |
| sarashina2.2-0.5b-instruct-v0.1 (Base) | 0.2457 | 0.3394 | 0.3635 |
| HACHI-Summary-Ja | 0.2757 | 0.3644 | 0.4044 |
1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
3
4# ===== settings =====
5
6# "","三行","100", "300", "500"から要約条件を設定
7SUMMARY_MODE = ""
8
9# 要約したいテキスト
10TEXT = """忠犬ハチ公
11
12犬種は秋田犬(あきたいぬ)で、性別はオス。名前はハチ。ハチ公の愛称でも知られる。
13ハチが飼い主を待ち続けた渋谷駅の出入り口の前には、ハチの銅像が設置されており、この「忠犬ハチ公像」は、渋谷のシンボルとして、観光名所としても有名である。
14
15ハチは、飼い主が死去した後も駅前で帰りを待ち続けた「忠犬」として知られる。東京・渋谷をはじめ、ゆかりの地には像が置かれている。
16特に、渋谷駅前のハチ公銅像は、いつしか待ち合わせの目印として使われるようになり、その銅像周囲は待ち合わせ場所としては「ハチ公前」などと呼ばれ、広く親しまれている。
17
18ハチの飼い主は、東京府豊多摩郡渋谷町大向(現・東京都渋谷区松濤一丁目)に住んでいた、東京帝国大学の教授・上野英三郎であった。
19彼は、大変な愛犬家であり、ハチの前にもたくさんの犬を飼っていた。出かける時には、渋谷駅までハチを伴うことも多かった。
20しかしながら、ハチを飼い始めた翌年にあたる1925年(大正14年)5月21日に上野は急死した。
21
22上野の死後も、駅前で亡くなった飼い主の帰りを毎日待ち続けたハチの姿は、新聞記事に掲載され、人々に感銘を与えたことから「忠犬ハチ公」と呼ばれるようになった。
23"""
24# 出典:忠犬ハチ公(https://ja.wikipedia.org/wiki/%E5%BF%A0%E7%8A%AC%E3%83%8F%E3%83%81%E5%85%AC)
25
26# ===== main =====
27
28SUMMARY_PROMPT_MAP = {
29 "100": "原文を100字程度で簡潔に要約してください。",
30 "300": "原文を300字程度で簡潔に要約してください。",
31 "500": "原文を500字程度で簡潔に要約してください。",
32 "三行": "原文を三行で簡潔に要約してください。",
33}
34
35SYSTEM_PROMPT = SUMMARY_PROMPT_MAP.get(
36 SUMMARY_MODE,
37 "原文を簡潔に要約してください。"
38)
39
40model_name = "hachi-intelligence/HACHI-Summary-Ja-sarashina2.2-0.5b-instruct-v0.1"
41
42tokenizer = AutoTokenizer.from_pretrained(
43 model_name,
44)
45
46model = AutoModelForCausalLM.from_pretrained(
47 model_name,
48 torch_dtype=torch.bfloat16,
49 device_map="auto",
50)
51
52chat_pipeline = pipeline(
53 "text-generation",
54 model=model,
55 tokenizer=tokenizer
56)
57
58messages = [
59 {"role": "system", "content": SYSTEM_PROMPT},
60 {"role": "user", "content": TEXT}
61]
62
63outputs = chat_pipeline(
64 messages,
65 do_sample=False,
66 # temperature=0.1,
67 repetition_penalty=1.00,
68 max_new_tokens=512,
69 num_return_sequences=1,
70)
71
72generated_messages = outputs[0]["generated_text"]
73summary = generated_messages[-1]["content"]
74
75
76# ===== result =====
77print("\n===== SYSTEM PROMPT =====")
78print(SYSTEM_PROMPT)
79print("===== SOURCE TEXT =====")
80print(TEXT,f"({len(TEXT)}文字)")
81print("\n===== SUMMARY =====")
82print(summary,f"({len(summary)}文字)")
83
84# ===== output =====
85# 秋田犬(あきたいぬ)のオス・ハチは、飼い主が死去した後も渋谷駅の出入り口で帰りを待ち続けた「忠犬ハチ公」として知られる。東京・渋谷のハチ公銅像は渋谷のシンボルとして観光名所となり、待ち合わせの目印として「ハチ公前」と呼ばれる。飼い主は東京府豊多摩郡渋谷町大向(現・東京都渋谷区松濤一丁目)の東京帝国大学教授・上野英三郎で、ハチの前に多数の犬を飼っていた。1925年(大正14年)5月21日に上野が死去した後も、ハチは渋谷駅で 飼い主の帰りを待ち続け、新聞記事で「忠犬ハチ公」と称された。 (242文字)