alihan/textattack-nlp-transformer
1
0
Fork 0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00
Code Issues Projects Releases Wiki Activity

add thought vectors

Browse Source
This commit is contained in:
Jack Morris
2020-05-06 11:16:17 -04:00
parent 3602625e0b
commit d647133230
5 changed files with 69 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
textattack/models/helpers/lstm_for_classification.py
View File

@@ -5,7 +5,7 @@ import torch.nn as nn
from textattack.shared import utils
from textattack.models.helpers import GloveEmbeddingLayer
from textattack.models.helpers.helper_utils import load_cached_state_dict
from textattack.models.helpers.utils import load_cached_state_dict
class LSTMForClassification(nn.Module):
""" A long short-term memory neural network for text classification.