https://github.com/gibrano/chatbot
Chatbot implementation with LSTM on tensorflow2.0 and multi-gpu capability.
https://github.com/gibrano/chatbot
Last synced: 6 months ago
JSON representation
Chatbot implementation with LSTM on tensorflow2.0 and multi-gpu capability.
- Host: GitHub
- URL: https://github.com/gibrano/chatbot
- Owner: gibrano
- License: mit
- Created: 2019-10-24T19:46:21.000Z (about 6 years ago)
- Default Branch: master
- Last Pushed: 2019-11-14T00:24:20.000Z (almost 6 years ago)
- Last Synced: 2024-11-13T03:34:51.236Z (12 months ago)
- Language: Python
- Size: 5.86 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
- Awesome-Tensorflow2 - gibrano/chatbot
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?")
```