search

namisan / mt-dnn

Multi-Task Deep Neural Networks for Natural Language Understanding
MIT License
2.22k stars 412 forks source link

How to train ny own dataset with MT-DNN #215

Closed ammaarahmad1999 closed 2 years ago

ammaarahmad1999 commented 3 years ago

I have a dataset for 3 tasks in csv file. How to train the model

namisan commented 2 years ago

please refer: prepro data: https://github.com/namisan/mt-dnn/blob/master/tutorials/Run_Your_Own_Task_in_MT-DNN.ipynb multi-task training: https://github.com/namisan/mt-dnn/blob/master/scripts/run_mt_dnn.sh