A Python package that combines the power of Keras with Gemini for natural language-driven neural network building.
- Python
- Keras
- Gemini API
- NLTK
To get a local copy up and running follow these simple steps.
- Python 3.x
pip install keras-gemini
from keras_gemini import prompt_to_keras
model = prompt_to_keras("Build a 3-layer sequential model")
if model:
model.summary()
To run these examples, users simply need to navigate to the examples/ directory and run any of the scripts. For example:
python examples/build_basic_model.py
- Natural Language Model Building: Build Keras sequential models by simply describing the desired architecture in natural language. For example:
Build a 3-layer sequential model
-
Automatic Model Compilation: The package automatically compiles the generated Keras model with default settings (optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy']).
-
Seamless Integration with Gemini: The
KerasGemini()
integrates directly into your Gemini chatbot flow, allowing for natural conversational model building.
-
Support for More Layer Types: Add support for a wider range of Keras layers (Convolutional, Recurrent, etc.) to enable building diverse network architectures.
-
Customizable Layer Parameters: Allow users to specify layer parameters (activation functions, number of units, etc.) through natural language prompts.
-
Advanced NLP for Model Understanding: Implement more robust natural language processing techniques to better extract user intent and complex model specifications.
-
Model Training and Evaluation: Provide functionality to train and evaluate the generated Keras models directly within the Gemini conversation.
-
Model Persistence: Allow users to save and load their custom-built models for later use.
-
Interactive Model Building: Enable users to iteratively refine their models by adding or removing layers, modifying parameters, and getting feedback in real-time.
Contributions are what make the open-source community such an amazing place to learn, inspire, and create. Any contributions you make are greatly appreciated.
- Fork the Project
- Create your Feature Branch (
git checkout -b feature/AmazingFeature
) - Commit your Changes (
git commit -m 'Add some AmazingFeature'
) - Push to the Branch (
git push origin feature/AmazingFeature
) - Open a Pull Request
Distributed under the MIT License. See License for more information.