Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,29 @@
|
|
1 |
-
---
|
2 |
-
license: mit
|
3 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: mit
|
3 |
+
---
|
4 |
+
# EEG Forecasting with Llama 3.1-8B and Time-LLM
|
5 |
+
|
6 |
+
This repository contains the code and model for forecasting EEG signals by combining the quantized Llama 3.1-8B model from [Hugging Face](https://huggingface.co/akshathmangudi/llama3.1-8b-quantized) and a modified version of the [Time-LLM](https://github.com/KimMeen/Time-LLM) framework.
|
7 |
+
|
8 |
+
## Overview
|
9 |
+
|
10 |
+
This project aims to leverage large language models (LLMs) for time-series forecasting, specifically focusing on EEG data. The integration of Llama 3.1-8B allows us to apply powerful sequence modeling capabilities to predict future EEG signal patterns with high accuracy and efficiency.
|
11 |
+
|
12 |
+
### Key Features
|
13 |
+
|
14 |
+
- **Quantized Llama 3.1-8B Model**: Utilizes a quantized version of Llama 3.1-8B to reduce computational requirements while maintaining performance.
|
15 |
+
- **Modified Time-LLM Framework**: Adapted the Time-LLM framework for EEG signal forecasting, allowing for efficient processing of EEG time-series data.
|
16 |
+
- **Scalable and Flexible**: The model can be easily adapted to other time-series forecasting tasks beyond EEG data.
|
17 |
+
|
18 |
+
## Getting Started
|
19 |
+
|
20 |
+
### Prerequisites
|
21 |
+
|
22 |
+
Before you begin, ensure you have the following installed:
|
23 |
+
|
24 |
+
- Python 3.8+
|
25 |
+
- PyTorch
|
26 |
+
- Transformers (Hugging Face)
|
27 |
+
- Time-LLM dependencies (see the original [Time-LLM repository](https://github.com/KimMeen/Time-LLM))
|
28 |
+
|
29 |
+
|