File size: 3,120 Bytes
5cb68b1 defb607 a034a51 defb607 5cb68b1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 |
---
license: mit
datasets:
- lunaopenlabs/LunaAi-dataset
language:
- en
metrics:
- character
base_model:
- lunaopenlabs/LunaAi
new_version: lunaopenlabs/LunaAi
pipeline_tag: text-classification
library_name: adapter-transformers
tags:
- luna
- open
- labs
- LunaAi
- text
- classification
---
# Luna AI
Luna AI is an open-source AI model developed by Luna OpenLabs for text classification tasks. Leveraging the BERT architecture, this model is designed to classify text into predefined categories efficiently and accurately.
## Table of Contents
- [Features](#features)
- [Installation](#installation)
- [Dataset](#dataset)
- [Usage](#usage)
- [Training the Model](#training-the-model)
- [Saving and Loading the Model](#saving-and-loading-the-model)
- [Testing the Model](#testing-the-model)
- [Contributing](#contributing)
- [License](#license)
- [Contact](#contact)
## Features
- **Text Classification**: Classify text data into various categories.
- **Built on BERT**: Utilizes the powerful BERT architecture for natural language understanding.
- **Easy Integration**: Works seamlessly with Hugging Face Transformers library.
- **Open Source**: Available for anyone to use, modify, and distribute.
## Installation
### Prerequisites
- Python 3.7 or higher
- pip (Python package installer)
### Clone the Repository
To clone the repository, run the following command:
bash
git clone https://github.com/LunaOpenLabs/Luna-Ai.git
### Install Requirements
To install the required packages, use:
bash
pip install -r requirements.txt
### Dataset
Luna AI requires a dataset in CSV format with two columns: text and label. An example dataset is provided in the data/ directory.
### Example Dataset Structure
Here’s an example of how the dataset should be structured:
csv
text,label
"I love this product!",1
"This is the worst experience.",0
### Usage
Training the Model
To train the model, execute the following command:
bash
python training/train.py
This command will load the dataset from data/dataset.csv and initiate the training process.
### Saving and Loading the Model
After training, save the trained model using:
bash
python save_model.py
This will save the model and its tokenizer in the luna_ai_model directory.
### Testing the Model
To test the model with sample inputs, you can use the test_model.py script. Modify the sample_text variable in the script as needed.
### Run the test script with:
bash
python test_model.py
### Example Output
The model will output the predicted class for the provided sample text.
### Contributing
Contributions are welcome! If you have suggestions, improvements, or bug fixes, please follow these steps:
Fork the repository.
Create a new branch (git checkout -b feature-branch).
Make your changes and commit them (git commit -m 'Add some feature').
Push to the branch (git push origin feature-branch).
Open a pull request.
### License
This project is licensed under the MIT License. See the LICENSE file for details.
### Contact
For questions, suggestions, or feedback, feel free to contact the Luna OpenLabs team at [lunaopenlabs@outlook.com]. |