Skip to content

Commit cc79ec9

Browse files
committed
Update code to TensorFlow 1.14 version
1 parent 6c2dd75 commit cc79ec9

9 files changed

Lines changed: 12 additions & 10 deletions

File tree

‎ANN/train_ann.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -269,4 +269,4 @@ def validation_step(x_val, y_val, writer=None):
269269

270270

271271
if __name__ == '__main__':
272-
train_ann()
272+
train_ann()

‎CNN/train_cnn.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -272,4 +272,4 @@ def validation_step(x_val, y_val, writer=None):
272272

273273

274274
if __name__ == '__main__':
275-
train_cnn()
275+
train_cnn()

‎CRNN/train_crnn.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -273,4 +273,4 @@ def validation_step(x_val, y_val, writer=None):
273273

274274

275275
if __name__ == '__main__':
276-
train_crnn()
276+
train_crnn()

‎FastText/train_fast.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -271,4 +271,4 @@ def validation_step(x_val, y_val, writer=None):
271271

272272

273273
if __name__ == '__main__':
274-
train_fasttext()
274+
train_fasttext()

‎HAN/train_han.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -271,4 +271,4 @@ def validation_step(x_val, y_val, writer=None):
271271

272272

273273
if __name__ == '__main__':
274-
train_han()
274+
train_han()

‎RCNN/train_rcnn.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -273,4 +273,4 @@ def validation_step(x_val, y_val, writer=None):
273273

274274

275275
if __name__ == '__main__':
276-
train_rcnn()
276+
train_rcnn()

‎RNN/train_rnn.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -271,4 +271,4 @@ def validation_step(x_val, y_val, writer=None):
271271

272272

273273
if __name__ == '__main__':
274-
train_rnn()
274+
train_rnn()

‎SANN/train_sann.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -273,4 +273,4 @@ def validation_step(x_val, y_val, writer=None):
273273

274274

275275
if __name__ == '__main__':
276-
train_sann()
276+
train_sann()

‎utils/data_helpers.py‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -309,7 +309,7 @@ def load_word2vec_matrix(word2vec_file):
309309
def data_word2vec(input_file, num_labels, word2vec_model):
310310
"""
311311
Create the research data tokenindex based on the word2vec model file.
312-
Return the class Data (includes the data tokenindex and data labels).
312+
Return the class _Data() (includes the data tokenindex and data labels).
313313
314314
Args:
315315
input_file: The research data
@@ -478,7 +478,9 @@ def load_data_and_labels(data_file, num_labels, word2vec_file, data_aug_flag):
478478
word2vec_file: The word2vec model file
479479
data_aug_flag: The flag of data augmented
480480
Returns:
481-
The class Data
481+
The class _Data()
482+
Raises:
483+
IOError: If word2vec model file doesn't exist
482484
"""
483485
# Load word2vec file
484486
if not os.path.isfile(word2vec_file):

0 commit comments

Comments
 (0)