Browse Source

remove datapath

tags/v0.5.0
wyg 5 years ago
parent
commit
95d9a76722
1 changed files with 1 additions and 11 deletions
  1. +1
    -11
      reproduction/text_classification/train_char_cnn.py

+ 1
- 11
reproduction/text_classification/train_char_cnn.py View File

@@ -22,17 +22,7 @@ class Config():
bert_embedding_larers= '4,-2,-1' bert_embedding_larers= '4,-2,-1'
train_epoch= 100 train_epoch= 100
num_classes=2 num_classes=2
task= "sst-2"
#yelp_p
datapath = {"train": "/remote-home/ygwang/yelp_polarity/train.csv",
"test": "/remote-home/ygwang/yelp_polarity/test.csv"}
#IMDB
#datapath = {"train": "/remote-home/ygwang/IMDB_data/train.csv",
# "test": "/remote-home/ygwang/IMDB_data/test.csv"}
# sst
# datapath = {"train": "/remote-home/ygwang/workspace/GLUE/SST-2/train.tsv",
# "dev": "/remote-home/ygwang/workspace/GLUE/SST-2/dev.tsv"}

task= "yelp_p"
lr=0.01 lr=0.01
batch_size=128 batch_size=128
model_size="large" model_size="large"


Loading…
Cancel
Save