wide-and-deep 1.0.0

Use wide and deep model to do multi-classification tasks.

wide_and_deep

Use wide and deep model to do multi-classification tasks.

This model is developed for the debt-collection robot. The robot calls customers within several topics. However, the human agents make different performances in these topics based on the payback money. If we find in a subtopic, human agents make a better performance, we will analyze the conversations they make with customers and write specific strategies for robots.

We collect the conversation between robot and customers and transform them into word2vector using a pre-trained model, use a bidirectional LSTM framework to extract features and use the wide and deep model to do the multi-label classification for higher accuracy (88-93%).

Install

pip install wide_and_deep

How to use

See demo https://jiaxiangbu.github.io/user_communication_classificationEX/