Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/gibrano/chatbot

Chatbot implementation with LSTM on tensorflow2.0 and multi-gpu capability.
https://github.com/gibrano/chatbot

Last synced: 2 months ago
JSON representation

Chatbot implementation with LSTM on tensorflow2.0 and multi-gpu capability.

Awesome Lists containing this project

README

        

# chatbot
Chatbot implementation with LSTM on tensorflow2.0 and multi-gpu capability. This implementation is based on the keras tutorial https://blog.keras.io/a-ten-minute-introduction-to-sequence-to-sequence-learning-in-keras.html.

# Train model

```
from modules.train import ChatBotModel

model = ChatBotModel()

if __name__ == '__main__':
model.train()
```

# Test model

```
from modules.chatbot import ChatBot

chatbot = ChatBot()

if __name__ == '__main__':
chatbot.run("Hello, how are you?")
```