Skip to content

XpressAI/xai-tensorflow-keras

Repository files navigation

Component LibrariesProject Templates
DocsInstallTutorialsDeveloper GuidesContributeBlogDiscord

Xircuits Component Library to interface with TensorFlow Keras! Build and deploy deep learning solutions with ease.


Xircuits Component Library for TensorFlow Keras

Integrates TensorFlow Keras into Xircuits for streamlined deep learning workflows. Simplifies model design, training, and deployment with reusable components. Reduces complexity and accelerates development for scalable AI solutions.

Table of Contents

Preview

TFKerasTransferLearning Example

TFKerasTransferLearning

TFKerasTransferLearning Result

TFKerasTransferLearning

TFKerasTrainImageClassifier Example

TFKerasTrainImageClassifier

TFKerasTrainImageClassifier Result

TFKerasTrainImageClassifier_Result

TFKerasModelPredict Example

TFKerasModelPredict

TFKerasModelPredict Result

TFKerasModelPredict_result

Prerequisites

Before you begin, you will need the following:

  1. Python3.9+.
  2. Xircuits.

Main Xircuits Components

KerasCreate1DInputModel:

Creates 1D Keras models for datasets with 1D input structures.

KerasCreate1DInputModel

KerasCreate2DInputModel:

Assembles 2D Keras models, ideal for image-based datasets.

KerasCreate2DInputModel

KerasTransferLearningModel:

Fetches TensorFlow Keras Models for transfer learning.

KerasTransferLearningModel

KerasTrainImageClassifier:

Trains Keras models for image classification tasks.

KerasEvaluateAccuracy:

Evaluates Keras models for accuracy and loss using test datasets.

TrainKerasModel:

Trains compiled Keras models with training data.

Try The Examples

We have provided an example workflow to help you get started with the TensorFlow Keras component library. Give it a try and see how you can create custom TensorFlow Keras components for your applications.

TFKerasTransferLearning

Demonstrates transfer learning with MobileNetV2 pre-trained on ImageNet to classify the "cats_vs_dogs" dataset, evaluate accuracy, and save the model.

TFKerasTrainImageClassifier

This workflow trains a 2D convolutional model on a dataset, evaluates its accuracy, and demonstrates basic image classification.

TFKerasModelPredict

Uses the pre-trained ResNet50 model to classify an input image and predict its class label.

Installation

To use this component library, ensure that you have an existing Xircuits setup. You can then install the TensorFlow Keras library using the component library interface, or through the CLI using:

xircuits install tensorflow_keras

You can also do it manually by cloning and installing it:

# base Xircuits directory
git clone https://github.com/XpressAI/xai-tensorflow-keras xai_components/xai_tensorflow_keras
pip install -r xai_components/xai_tensorflow_keras/requirements.txt 

About

Xircuits component library for Tensorflow Keras

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Contributors 3

  •  
  •  
  •  

Languages