jonad / identify_question_pair_with_the_same_intent Goto Github PK

• Python 3.6
• Gensim
• Keras 2.0
• NLTK
• xgboost

• Training data

Navigate to notebooks and run the following command

jupyter notebook data_preprocessing.ipynb

After running all cells in the notebooks, you will have a preprocessed data into the data folder

Train either the xgboost model or the cnn model by running the following command from the project's root directory

python train_cnn.py

This will train the neural network model. You can modify this file to run the model with different parameters

python train_xgb.py

This will train the xgboost model. You can modify this file to run the model with different parameter.