Algorithm, that identifies whether given text from fiction writing, technical literature or сolloquialism
You could see how model trained in Train.ipynb
If you want to test it, you need first get model parameters. To do so either run Train.ipynb or download them directly from https://drive.google.com/drive/folders/1Wb8vsdFYgm73K4eR8gP-lyl5MinDSdo-?usp=sharing Then just run
python identify_style.py 'Do you remember when he used to come to the Terrace? I wanted to take him fishing but I was too timid to ask him. Then I asked you to ask him and you were too timid. I would like to take the great DiMaggio fishing'