-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrequirements.txt
More file actions
19 lines (18 loc) · 826 Bytes
/
requirements.txt
File metadata and controls
19 lines (18 loc) · 826 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
pandas
pydantic # to enforce schema for LLM outputs
typing # to enforce schema for LLM outputs
ollama
tqdm # progress bar for each experiment
PyYAML
scikit-learn # to get the classification metrics, classification_report, confusion_matrix
seaborn # to visualise results in Jupyter notebook
matplotlib # to visualise results in Jupyter notebook
python-dotenv # for loading environment variables from .env file
openai # for Nebius and OpenAI API calls
tenacity # for API model retries
google-genai # for Gemini API calls
# # Below: libraries for embedding approaches
# huggingface-hub
# datasets # to convert datasets from TSV to Hugging Face Dataset format
# transformers # to use Hugging Face Transformers eg BERT
# torch # to use PyTorch to train and evaluate BERT