cerebrate-sdk 0.2.1

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

Reusing presets

from cerebrate_sdk import Cerebrate

c = Cerebrate('YOUR_API_KEY')

# key-value pairs
# variable_name: value
# variables are defined in UI during preset creation
variables = {
    "stock": "tennis balls, tennis rockets, soccer balls, soccer uniform, basketballs, tennis uniform, tents, backpacks, socks",
    "cart": "tennis ball, socks"
}

result = c.preset_predict(preset_id="YOUR_PRESET_ID", variables=variables)

print(result[0])