cerebrate-sdk 0.2.0

Cerebrate SDK

Cerebrate SDK

Install

with poetry

poetry add cerebrate-sdk

or with pip

pip install cerebrate-sdk

Examples

Fake email detector

from cerebrate_sdk import Cerebrate

c = Cerebrate('YOUR_API_KEY')

task = "Detect if email is fake or real"
examples = [
    "[e-mail]: fake"
    "[e-mail]: real",
]

result = c.predict(task, examples, "[e-mail]: ")

print(result[0])
# fake

With options

from cerebrate_sdk import Cerebrate, Options

c = Cerebrate('YOUR_API_KEY')

options = Options(
    stop=['Q:'],
    temperature=0.7,
    max_tokens=100,
    top_p=1,
    presence_penalty=0,
    frequency_penalty=0,
    best_of=1
)

task = "Detect if email is fake or real"
examples = [
    "[e-mail]: fake"
    "[e-mail]: real",
]

result = c.predict(task, examples, "[e-mail]: ", options=options)

print(result[0])
# fake

Raw usage

from cerebrate_sdk import Cerebrate

c = Cerebrate("YOUR_API_KEY")

result = c.raw("Suggest the next item for user's cart."
               "Cart: bacon, eggs, tomatoes"
               "Suggested item: ")
print(result[0])
# sausage