aexol-studio/recommendation_content_based – Run with an API on Replicate

aexol-studio / recommendation_content_based

Cold

Public
22 runs
L40S

Iterate in playground

Run with an API

Playground API Examples README Versions

Input

reviews

string

Shift + Return to add a new line

api:McAuley-Lab/Amazon-Reviews-2023 raw_review_Digital_Musicapi:McAuley-Lab/Amazon-Reviews-2023 raw_review_Digital_Music

HuggingFace Datasets API (api:McAuley-Lab/Amazon-Reviews-2023 raw_meta_Digital_Music) or path (path:data/reviews.csv)

Default: "api:McAuley-Lab/Amazon-Reviews-2023 raw_review_Digital_Music"

user_id

string

Shift + Return to add a new line

user id

Default: "AHZZPUYPNZQ7QXK55HGGE3Z7TTEA"

user_column

string

Shift + Return to add a new line

user column

Default: "user_id"

product_column

string

Shift + Return to add a new line

product column

Default: "parent_asin"

name_column

string

Shift + Return to add a new line

title column

Default: "title"

text_column

string

Shift + Return to add a new line

text column

Default: "text"

top_n

integer

top n recommendation

Default: 10

pretrained

boolean

Whether to use a pretrained model

Default: false

show_metrics

boolean

Whether to show the evaluation metrics

Default: false

Run this model in Node.js with one line of code:

npx create-replicate --model=aexol-studio/recommendation_content_based

or set up a project from scratch

Install Replicate’s Node.js client library:

npm install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import and set up the client:

import Replicate from "replicate";

const replicate = new Replicate({
  auth: process.env.REPLICATE_API_TOKEN,
});

Run aexol-studio/recommendation_content_based using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "aexol-studio/recommendation_content_based:a23e55f1956d6b573f7addfa00bb756d9998e14d726b5c9e0bfcb4ec8bb7cfc8",
  {
    input: {
      top_n: 10,
      reviews: "api:McAuley-Lab/Amazon-Reviews-2023 raw_review_Digital_Music",
      user_id: "AHZZPUYPNZQ7QXK55HGGE3Z7TTEA",
      pretrained: false,
      name_column: "title",
      text_column: "text",
      user_column: "user_id",
      show_metrics: false,
      product_column: "parent_asin"
    }
  }
);

console.log(output);

To learn more, take a look at the guide on getting started with Node.js.

Install Replicate’s Python client library:

pip install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import the client:

import replicate

Run aexol-studio/recommendation_content_based using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

output = replicate.run(
    "aexol-studio/recommendation_content_based:a23e55f1956d6b573f7addfa00bb756d9998e14d726b5c9e0bfcb4ec8bb7cfc8",
    input={
        "top_n": 10,
        "reviews": "api:McAuley-Lab/Amazon-Reviews-2023 raw_review_Digital_Music",
        "user_id": "AHZZPUYPNZQ7QXK55HGGE3Z7TTEA",
        "pretrained": False,
        "name_column": "title",
        "text_column": "text",
        "user_column": "user_id",
        "show_metrics": False,
        "product_column": "parent_asin"
    }
)
print(output)

To learn more, take a look at the guide on getting started with Python.

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Run aexol-studio/recommendation_content_based using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

curl -s -X POST \
  -H "Authorization: Bearer $REPLICATE_API_TOKEN" \
  -H "Content-Type: application/json" \
  -H "Prefer: wait" \
  -d $'{
    "version": "aexol-studio/recommendation_content_based:a23e55f1956d6b573f7addfa00bb756d9998e14d726b5c9e0bfcb4ec8bb7cfc8",
    "input": {
      "top_n": 10,
      "reviews": "api:McAuley-Lab/Amazon-Reviews-2023 raw_review_Digital_Music",
      "user_id": "AHZZPUYPNZQ7QXK55HGGE3Z7TTEA",
      "pretrained": false,
      "name_column": "title",
      "text_column": "text",
      "user_column": "user_id",
      "show_metrics": false,
      "product_column": "parent_asin"
    }
  }' \
  https://api.replicate.com/v1/predictions

To learn more, take a look at Replicate’s HTTP API reference docs.

Output

No output yet! Press "Submit" to start a prediction.

Run time and cost

This model runs on Nvidia L40S GPU hardware. We don't yet have enough runs of this model to provide performance information.

Readme

This model doesn't have a readme.