Browse Source

text_classfication

tags/v0.4.10
lyhuang18 5 years ago
parent
commit
46c82a7daa
4 changed files with 6 additions and 6 deletions
  1. BIN
      reproduction/text_classification/results_LSTM.xlsx
  2. +2
    -2
      reproduction/text_classification/train_awdlstm.py
  3. +2
    -2
      reproduction/text_classification/train_lstm.py
  4. +2
    -2
      reproduction/text_classification/train_lstm_att.py

BIN
reproduction/text_classification/results_LSTM.xlsx View File


+ 2
- 2
reproduction/text_classification/train_awdlstm.py View File

@@ -33,9 +33,9 @@ class Config():


task_name = "IMDB" task_name = "IMDB"
datapath={"train":"IMDB_data/train.csv", "test":"IMDB_data/test.csv"} datapath={"train":"IMDB_data/train.csv", "test":"IMDB_data/test.csv"}
load_model_path="./result_IMDB/best_BiLSTM_SELF_ATTENTION_acc_2019-07-07-04-16-51"
save_model_path="./result_IMDB_test/" save_model_path="./result_IMDB_test/"
opt=Config

opt=Config()




# load data # load data


+ 2
- 2
reproduction/text_classification/train_lstm.py View File

@@ -30,9 +30,9 @@ class Config():


task_name = "IMDB" task_name = "IMDB"
datapath={"train":"IMDB_data/train.csv", "test":"IMDB_data/test.csv"} datapath={"train":"IMDB_data/train.csv", "test":"IMDB_data/test.csv"}
load_model_path="./result_IMDB/best_BiLSTM_SELF_ATTENTION_acc_2019-07-07-04-16-51"
save_model_path="./result_IMDB_test/" save_model_path="./result_IMDB_test/"
opt=Config

opt=Config()




# load data # load data


+ 2
- 2
reproduction/text_classification/train_lstm_att.py View File

@@ -32,9 +32,9 @@ class Config():


task_name = "IMDB" task_name = "IMDB"
datapath={"train":"IMDB_data/train.csv", "test":"IMDB_data/test.csv"} datapath={"train":"IMDB_data/train.csv", "test":"IMDB_data/test.csv"}
load_model_path="./result_IMDB/best_BiLSTM_SELF_ATTENTION_acc_2019-07-07-04-16-51"
save_model_path="./result_IMDB_test/" save_model_path="./result_IMDB_test/"
opt=Config

opt=Config()




# load data # load data


Loading…
Cancel
Save