GPT-3のExamples/Extract contact informationを試してみた

サイト内検索

カスタム検索

フリーソフト・オープンソースを活用しよう。

GPT-3のExamples/Extract contact informationを試してみた

2021/11/23

　OpenAIの自然言語処理モデルGPT-3のExamples/Extract contact informationを試してみました。

Extract contact information

debian:~$ python openai_examples_extract_contact_info.py

('このメールから連絡先を抽出して:\n\n田中さんへ,\n\nこの前言ってた東京都庁の住所は、これみたいですよ。\n\n〒163-8001 東京都新宿区西新宿2-8-1 電話：03-5321-1111[代表]\n\n取り急ぎ、ご連絡まで。\n\n佐藤\n\n名前と住所:\n',)

田中さんへ,

この前言ってた東京都庁の住所は、これみたいですよ。

〒163-8001 東京都新

debian:~$ python openai_examples_extract_contact_info.py

{

"id": "cmpl-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx",

"object": "text_completion",

"created": xxxxxxxx,

"model": "davinci:2020-05-03",

"choices": [

{

"text": " \n田中さんへ,\n\nこの前言ってた東京都庁の住所は、これみたいですよ。\n\n〒163-8001 東京都新",

"index": 0,

"logprobs": null,

"finish_reason": "length"

}

]

}

debian:~$

　文中に含まれる連絡先情報を抽出するExtract contact information。

　英語の元サンプルでは、欠落していたようですが、宛先の名前も拾い、[都庁]なのか[住所]なのか、こうしたワードもヒットしてしまい文章も入っているものの、郵便番号も住所もバッチリ抽出してくれているようです。

　というか、行単位で抽出するようですね、そりゃそうか...。

　応答が、途切れていますが、[finish_reason]が[length]とあるのでスクリプト内の[max_tokens]値が許せば、全て抽出できていることでしょう。

　正しく抽出されたかを確認するためには、前半に住所と無縁の一文も入れるか、応答をもっと表示するようにすべきでしたね...。

debian:~$ cat openai_examples_extract_contact_info.py

import openai

import os

import json

openai.api_key = os.getenv("OPENAI_API_KEY")

#prompt="Extract the mailing address from this email:\n\nDear Kelly,\n\nIt was great to talk to you at the seminar. I thought Jane's talk was quite good.\n\nThank you for the book. Here's my address 2111 Ash Lane, Crestview CA 92002\n\nBest,\n\nMaya\n\nName and address:\n",

prompt="このメールから連絡先を抽出して:\n\n田中さんへ,\n\nこの前言ってた東京都庁の住所は、これみたいですよ。\n\n〒163-8001 東京都新宿区西新宿2-8-1 電話：03-5321-1111[代表]\n\n取り急ぎ、ご連絡まで。\n\n佐藤\n\n名前と住所:\n"

response = openai.Completion.create(

engine="davinci-instruct-beta",

prompt=prompt,

temperature=0,

max_tokens=64,

top_p=1.0,

frequency_penalty=0.0,

presence_penalty=0.0

)

#print(prompt)

print(str(prompt))