Yi-1.5-6B-Chat-Math
Yi-1.5-6B-Chat-Math is a specialized natural language processing model fine-tuned from Yi-1.5 to excel in mathematical problem-solving and related tasks. Designed to handle a wide range of mathematical queries, from algebra and geometry to calculus and beyond, this model serves as a powerful tool for students, educators, and researchers alike.
Features
- Mathematical Problem Solving: Accurately solves various types of mathematical problems, including but not limited to algebra, geometry, and calculus.
- Formula Derivation: Assists in deriving and explaining mathematical formulas to enhance understanding of complex concepts.
- Multilingual Support: Capable of handling mathematical queries in multiple languages, enhancing accessibility for a diverse user base.
- Custom Fine-Tuning: Trained on a proprietary dataset to ensure high performance and reliability in mathematical contexts.
Dataset
The model has been fine-tuned using a custom dataset tailored for advanced mathematical tasks. The dataset is openly available for research and development purposes.
- Dataset Name: Advanced-Math
- Access Link: Advanced-Math Dataset
Installation
To get started with Yi-1.5-6B-Chat-Math, ensure you have the necessary dependencies installed:
pip install transformers torch
Usage
Below is a simple example demonstrating how to use the model for solving a mathematical equation:
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
# Load the tokenizer and model
tokenizer = AutoTokenizer.from_pretrained("haijian06/Yi-1.5-6B-Chat-Math")
model = AutoModelForCausalLM.from_pretrained("haijian06/Yi-1.5-6B-Chat-Math", torch_dtype=torch.float16, device_map="auto")
input_text = "Solve the equation x^2 - 5x + 6 = 0 Let's solve this step-by-step:"
inputs = tokenizer(input_text, return_tensors="pt").to(model.device)
with torch.no_grad():
outputs = model.generate(
**inputs,
max_new_tokens=200,
do_sample=True,
temperature=0.7,
top_p=0.95,
)
answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(answer)
Model answer:
Solve the equation x^2 - 5 x + 6 = 0 Let's solve this step-by-step:
Step 1: Factor the equation
The equation can be factored as follows:
x^2 - 5x + 6 = 0
(x - 2)(x - 3) = 0
Step 2: Apply the zero product property
If the product of two numbers is zero, then at least one of the numbers must be zero.
So, either (x - 2) = 0 or (x - 3) = 0
Step 3: Solve for x
If (x - 2) = 0, then x = 2
If (x - 3) = 0, then x = 3
So, the solutions are x = 2 and x = 3.
Answer: 2, 3
Contributing
Contributions are welcome! Whether you have suggestions for improvements, bug reports, or want to contribute code, feel free to open an issue or submit a pull request on GitHub.
License
This project is licensed under the Apache-2.0 License.
Contact
For more information, support, or inquiries, please visit my GitHub profile:
- GitHub: https://github.com/Haijian06
- Downloads last month
- 11