|
|
|
@@ -43,7 +43,7 @@ def lstm_create_dataset(data_home, batch_size, repeat_num=1, training=True): |
|
|
|
|
|
|
|
def _convert_to_mindrecord(data_home, features, labels, weight_np=None, training=True): |
|
|
|
""" |
|
|
|
convert imdb dataset to mindrecoed dataset |
|
|
|
convert imdb dataset to mindrecord dataset |
|
|
|
""" |
|
|
|
if weight_np is not None: |
|
|
|
np.savetxt(os.path.join(data_home, 'weight.txt'), weight_np) |
|
|
|
@@ -76,7 +76,7 @@ def _convert_to_mindrecord(data_home, features, labels, weight_np=None, training |
|
|
|
|
|
|
|
def convert_to_mindrecord(embed_size, aclimdb_path, preprocess_path, glove_path): |
|
|
|
""" |
|
|
|
convert imdb dataset to mindrecoed dataset |
|
|
|
convert imdb dataset to mindrecord dataset |
|
|
|
""" |
|
|
|
parser = ImdbParser(aclimdb_path, glove_path, embed_size) |
|
|
|
parser.parse() |
|
|
|
|