first commit

README.md

@@ -31,7 +31,7 @@ PCNN+ATT_RA | 0.403 | **0.422**
 
 1. upzip the file `NYT_data/NYT_data.zip`
 
-2. Make data folder in the following structure
+2. make data folder in the following structure
 
 ```
 Intra-Bag-and-Inter-Bag-Attentions

train.py

@@ -7,13 +7,11 @@
 import numpy as np
 import torch
 import torch.nn as nn
-import torch.nn.functional as Fh
 import torch.optim as optim
 import torch.nn.functional as F
-import torch.nn.init as weight_init
 from sklearn.metrics import average_precision_score
 from torch.autograd import Variable
-from preprocess.dataset import DocumentContainer
+from preprocess.data2pkl import DocumentContainer
 from model.model_bagatt import Model