Skip to content
A deep learning engine for your own personal virtual assistant
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.github
animius
demos Add a notebook for demo. Aug 25, 2019
docs Update python.md in docs. Aug 3, 2019
.gitignore
LICENSE
README.md
mkdocs.yml Add am.Model to doc. Jun 17, 2019
setup.py

README.md




Animius is an open source software library for creating deep-learning-powered virtual assistants. It provides an intuitive workflow that extracts data from existing media (such as anime and TV shows) and trains on them to provide a personalized AI. The flexible architecture enables you to add custom functionality to your virtual assistant.

Animius also ships with a high-level API animius.Console that allows users without programming experience to use Animius.

Installation

Install the current release from PyPi:

pip install animius

Then, install Tensorflow (recommended version 1.14. Animius does not support TF 2.0). We recommend using the GPU package (tensorlfow-gpu) if you are going to train your own virtual assistant. Read more on Tensorflow installation here.

See Installing Animius for detailed instructions and Docker installation guide.

Getting Started

For more information, check out our quick start guides.

Here's a quick overview of how to create an intent-ner model with Animius:

# create a new model config for intent ner and set default values
myIntentModelConfig = am.ModelConfig(cls='IntentNER')
myIntentModelConfig.apply_defaults()

# we can set & view the values of the config before creating the model 
print(myModelIntentConfig.model_structure)
myIntentModelConfig.hyperparameters['learning_rate'] = 0.01

# load a pretrained word embedding from disk
myGloveEmbedding = am.WordEmbedding()
myGloveEmbedding.create_embedding('\\resources\\glove.twitter.27B.50d.txt', vocab_size=40000)

# create a data object for your training set
myIntentData = am.IntentNERData()
myIntentData.add_embedding_class(myGloveEmbedding)
myIntentData.set_intent_folder('resources\\IntentNER Data')

# create a model
myIntentModel = am.IntentNER.IntentNERModel()

# build a TF graph based on the config and data
myIntentModel.build_graph(myModelConfig, myData)
# start tensorflow
myIntentModel.init_tensorflow()

# train!
myIntentModel.train(epochs=10)

After training the intent-ner and chatbot models, we can create a waifu - a virtual assistant. The code is a bit long, so I will skip a lot of it. Feel free to check out our demo and our quick start guide for more information.

Anyways, you can simply interact with a working waifu like this:

myWaifu.predict('Hey!')

myWaifu.predict('What is the time in New York right now?')

myWaifu.predict('Play me some music')

For more information

Also, feel free to check out our other projects surrounding Animius:

License

Apache License 2.0

You can’t perform that action at this time.