 ai.google.dev에서 보기
|
 Google Colab에서 실행
|
 GitHub에서 소스 보기
|
이 노트북에서는 Google의 최신 대규모 언어 모델에 액세스할 수 있게 해주는 PaLM API를 시작하는 방법을 알아봅니다. 여기에서는 PaLM API의 텍스트 생성 기능을 사용하는 방법을 알아봅니다.
설정
먼저 PaLM API Python 라이브러리를 다운로드하고 설치합니다.
pip install -q google-generativeai
import pprint
import google.generativeai as palm
API 키 가져오기
시작하려면 API 키를 만들어야 합니다.
palm.configure(api_key='YOUR_API_KEY')
텍스트 생성
palm.list_models 함수를 사용하여 사용 가능한 모델을 찾습니다.
models = [m for m in palm.list_models() if 'generateText' in m.supported_generation_methods]
model = models[0].name
print(model)
models/text-bison-001
palm.generate_text 메서드를 사용하여 텍스트를 생성합니다.
prompt = """
You are an expert at solving word problems.
Solve the following problem:
I have three houses, each with three cats.
each cat owns 4 mittens, and a hat. Each mitten was
knit from 7m of yarn, each hat from 4m.
How much yarn was needed to make all the items?
Think about it step by step, and show your work.
"""
completion = palm.generate_text(
model=model,
prompt=prompt,
temperature=0,
# The maximum length of the response
max_output_tokens=800,
)
print(completion.result)
There are 3 houses * 3 cats / house = 9 cats. So, 9 cats * 4 mittens / cat = 36 mittens were made. Also, 9 cats * 1 hat / cat = 9 hats were made. So, 36 mittens * 7m / mitten = 252m of yarn was used for the mittens. Also, 9 hats * 4m / hat = 36m of yarn was used for the hats. In total, 252m + 36m = 288m of yarn was used. Thus, the answer is 288.
옵션 더보기
palm.generate_text 함수에는 언급할 만한 다른 인수가 몇 가지 있습니다.
중지 시퀀스
생성을 조기에 중지하려면 stop_sequences 인수를 사용하세요.
예를 들어 LLM은 종종 산술을 실수합니다. <calc> 태그에 방정식을 입력하여 모델에 '계산기를 사용'하도록 요청할 수 있습니다.
프롬프트를 수정할 수 있도록 모델을 닫는 태그에서 중지하도록 합니다.
calc_prompt = f"""
Please solve the following problem.
{prompt}
----------------
Important: Use the calculator for each step.
Don't do the arithmetic in your head.
To use the calculator wrap an equation in <calc> tags like this:
<calc> 3 cats * 2 hats/cat </calc> = 6
----------------
"""
equation=None
while equation is None:
completion = palm.generate_text(
model=model,
prompt=calc_prompt,
stop_sequences=['</calc>'],
# The maximum length of the response
max_output_tokens=800,
)
try:
response, equation = completion.result.split('<calc>', maxsplit=1)
except Exception:
continue
print(response)
Chain-of-thought: There are three houses, and each house has three cats, so there are 3 houses * 3 cats / house = 9 cats. Each cat has 4 mittens, so the cats need 9 cats * 4 mittens / cat = 36 mittens. Each mitten takes 7m of yarn, so 36 mittens * 7m / mitten = 252m of yarn. Each cat has a hat, and each hat takes 4m of yarn, so 9 cats * 4m / cat = 36m of yarn. So, in total, 36m + 252m = 288m of yarn were needed. The answer should be
print(equation)
9 cats * 4 mittens / cat
여기에서 결과를 계산하고 계속 진행할 모델의 새 프롬프트를 조합할 수 있습니다. 완전히 작동하는 구현은 텍스트 계산기 예를 참고하세요.
후보자
일반적으로 LLM에서 생성되는 텍스트에는 어느 정도의 임의성이 있습니다. (LLM 기본 지침서에서 그 이유를 자세히 알아보세요.) 즉, 동일한 입력으로 API를 두 번 이상 호출하면 다른 응답을 받을 수 있습니다. 이 기능을 유용하게 사용하여 대체 모델 응답을 받을 수 있습니다.
temperature 인수는 응답의 분산을 제어합니다. palm.Model 객체는 temperature 및 기타 매개변수의 기본값을 제공합니다.
models[0]
Model(name='models/text-bison-001', base_model_id='', version='001', display_name='Text Bison', description='Model targeted for text generation.', input_token_limit=8196, output_token_limit=1024, supported_generation_methods=['generateText'], temperature=0.7, top_p=0.95, top_k=40)
candidate_count 인수는 반환되는 응답 수를 제어합니다.
completion = palm.generate_text(
model=model,
prompt=prompt,
# The number of candidates to return
candidate_count=8,
# Set the temperature to 1.0 for more variety of responses.
temperature=1.0,
max_output_tokens=800,
)
print(completion.result)
In each house there are 3 cats * 4 mittens / cat = 12 mittens. In total there are 3 houses * 12 mittens / house = 36 mittens. In total there are 36 mittens * 7m / mitten = 252m of yarn for the mittens. In total there are 3 houses * 3 cats / house * 1 hat / cat = 9 hats. In total there are 9 hats * 4m / hat = 36m of yarn for the hats. In total there are 36m yarn for the hats + 252m yarn for the mittens = 288m of yarn. The answer: 288.
여러 후보를 요청해도 Completion.result 속성에는 여전히 첫 번째 후보만 포함됩니다. Completion.candidates 속성에는 다음 항목이 모두 포함됩니다.
import pprint
pprint.pprint(completion.candidates)
[{'output': 'In each house there are 3 cats * 4 mittens / cat = 12 mittens. In '
'total there are 3 houses * 12 mittens / house = 36 mittens. In '
'total there are 36 mittens * 7m / mitten = 252m of yarn for the '
'mittens. In total there are 3 houses * 3 cats / house * 1 hat / '
'cat = 9 hats. In total there are 9 hats * 4m / hat = 36m of yarn '
'for the hats. In total there are 36m yarn for the hats + 252m '
'yarn for the mittens = 288m of yarn.\n'
'The answer: 288.',
'safety_ratings': [{'category': <HarmCategory.HARM_CATEGORY_DEROGATORY: 1>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_TOXICITY: 2>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_VIOLENCE: 3>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_SEXUAL: 4>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_MEDICAL: 5>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_DANGEROUS: 6>,
'probability': <HarmProbability.NEGLIGIBLE: 1>}]},
{'output': 'Each house has 3 cats, so each house needs 3 * 4 = 12 mittens. '
"With three houses, that's 3 * 12 = 36 mittens. And each house "
'needs 3 * 1 = 3 hats. So in total, we need 3 hats + 36 mittens = '
'39 items. Each mitten needs 7 meters of yarn, so 39 mittens need '
'39 * 7 = 273 meters of yarn. Each hat needs 4 meters of yarn, and '
"we need 3 hats, so that's 4 * 3 = 12 meters of yarn. So in total, "
'we needed 12 + 273 = 285 meters of yarn.\n'
'Thus, the answer is 285.',
'safety_ratings': [{'category': <HarmCategory.HARM_CATEGORY_DEROGATORY: 1>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_TOXICITY: 2>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_VIOLENCE: 3>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_SEXUAL: 4>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_MEDICAL: 5>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_DANGEROUS: 6>,
'probability': <HarmProbability.NEGLIGIBLE: 1>}]},
{'output': 'There are 3 houses * 3 cats / house = 9 cats. There are 9 cats * '
'4 mittens / cat = 36 mittens. There are 9 cats * 1 hat / cat = 9 '
'hats. The total amount of yarn for the mittens is 36 mittens * 7m '
'/ mitten = 252m. The total amount of yarn for the hats is 9 hats '
'* 4m / hat = 36m. The total amount of yarn is 252m + 36m = 288m.\n'
'Thus, the answer is 288.',
'safety_ratings': [{'category': <HarmCategory.HARM_CATEGORY_DEROGATORY: 1>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_TOXICITY: 2>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_VIOLENCE: 3>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_SEXUAL: 4>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_MEDICAL: 5>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_DANGEROUS: 6>,
'probability': <HarmProbability.NEGLIGIBLE: 1>}]},
{'output': 'There are 3 houses * 3 cats / house = 9 cats. Each cat has 4 '
'mittens + 1 hat = 5 items. So the total number of items is 9 cats '
'* 5 items / cat = 45 items. Thus, 45 items * 7m / item = 315m of '
'yarn was needed.\n'
'Thus, the answer is 315.',
'safety_ratings': [{'category': <HarmCategory.HARM_CATEGORY_DEROGATORY: 1>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_TOXICITY: 2>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_VIOLENCE: 3>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_SEXUAL: 4>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_MEDICAL: 5>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_DANGEROUS: 6>,
'probability': <HarmProbability.NEGLIGIBLE: 1>}]},
{'output': 'Chain-of-thought:\n'
'There are 3 houses * 3 cats / house = 9 cats.\n'
'The cats need 9 cats * 4 mittens / cat = 36 mittens.\n'
'The cats need 9 cats * 1 hat / cat = 9 hats.\n'
'The mittens need 36 mittens * 7m / mitten = 252m of yarn.\n'
'The hats need 9 hats * 4m / hat = 36m of yarn.\n'
'Therefore, the total amount of yarn needed is 252m + 36m = 288m.\n'
'\n'
'The answer should be 288',
'safety_ratings': [{'category': <HarmCategory.HARM_CATEGORY_DEROGATORY: 1>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_TOXICITY: 2>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_VIOLENCE: 3>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_SEXUAL: 4>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_MEDICAL: 5>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_DANGEROUS: 6>,
'probability': <HarmProbability.NEGLIGIBLE: 1>}]},
{'output': 'First find the total number of cats: 3 houses * 3 cats / house = '
'9 cats. Then multiply that number by the number of mittens per '
'cat to find the total number of mittens: 9 cats * 4 mittens / cat '
'= 36 mittens. Then multiply that number by the number of meters '
'of yarn per mitten to find the total amount of yarn used for '
'mittens: 36 mittens * 7 meters / mitten = 252 meters. Then do the '
'same thing for hats: 9 cats * 1 hat / cat = 9 hats. Then multiply '
'that number by the number of meters of yarn per hat to find the '
'total amount of yarn used for hats: 9 hats * 4 meters / hat = 36 '
'meters. Then add the amount of yarn used for mittens and hats to '
'find the total amount of yarn used: 36 meters + 252 meters = 288 '
'meters.\n'
'Thus, the answer is 288.',
'safety_ratings': [{'category': <HarmCategory.HARM_CATEGORY_DEROGATORY: 1>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_TOXICITY: 2>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_VIOLENCE: 3>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_SEXUAL: 4>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_MEDICAL: 5>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_DANGEROUS: 6>,
'probability': <HarmProbability.NEGLIGIBLE: 1>}]},
{'output': 'The total number of cats is 3 houses * 3 cats / house = 9 cats. '
'So, the total number of mittens is 9 cats * 4 mittens / cat = 36 '
'mittens. The total number of hats is 9 cats * 1 hat / cat = 9 '
'hats. The total length of yarn needed to make the mittens is 36 '
'mittens * 7 m / mitten = 252 m. The total length of yarn needed '
'to make the hats is 9 hats * 4 m / hat = 36 m. So, the total '
'length of yarn needed is 252 m + 36 m = 288 m.\n'
'\n'
'The answer: 288',
'safety_ratings': [{'category': <HarmCategory.HARM_CATEGORY_DEROGATORY: 1>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_TOXICITY: 2>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_VIOLENCE: 3>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_SEXUAL: 4>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_MEDICAL: 5>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_DANGEROUS: 6>,
'probability': <HarmProbability.NEGLIGIBLE: 1>}]},
{'output': 'There are 3 houses with 3 cats each, so 3 * 3 = 9 cats. Each cat '
'has 4 mittens and a hat, so 9 cats * 4 mittens / cat + 9 cats * 1 '
'hat / cat = 36 mittens and 9 hats. Each mitten takes 7m of yarn '
'and each hat takes 4m of yarn, so the total yarn needed is 36 '
'mittens * 7m / mitten + 9 hats * 4m / hat = 252m + 36m = 288m.\n'
'The answer: 288.',
'safety_ratings': [{'category': <HarmCategory.HARM_CATEGORY_DEROGATORY: 1>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_TOXICITY: 2>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_VIOLENCE: 3>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_SEXUAL: 4>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_MEDICAL: 5>,
'probability': <HarmProbability.NEGLIGIBLE: 1>},
{'category': <HarmCategory.HARM_CATEGORY_DANGEROUS: 6>,
'probability': <HarmProbability.NEGLIGIBLE: 1>}]}]
이 문제의 답을 알고 있으므로 해결률을 쉽게 확인할 수 있습니다.
import numpy as np
np.mean(['288' in c['output'] for c in completion.candidates])
0.75