ML Classification vs LLM Classification

This demo repo compares a Pytorch RNN model vs an LLM for simple classification tasks.

RNN model source

Environment Setup

Copy the .env.example file to .env and fill in the values. Or export values in shell.

Set the OPENAI_API_KEY environment variable to access the OpenAI models.

Set the PAREA_API_KEY environment variable to access tracing with PareaAI:

or export values in shell:

export OPENAI_API_KEY= <YOUR OPENAI API KEY>
export PAREA_API_KEY= <YOUR OPENAI API KEY>

Get Started

pip install -r requirements.txt

Then run main/rnn_experiment.py

Results

View experiment results on Parea AI.