corrected readme file

This commit is contained in:
Siddharth 2019-06-18 22:17:43 +05:30
parent a86f074dca
commit 226cbc05fd

View File

@ -1,4 +1,4 @@
# networks.py # networks.py
## `class LSTM_TAN(nn.Module):` ## `class LSTM_TAN(nn.Module):`
``` ```
def __init__(self,version,embedding_dim, hidden_dim, vocab_size, n_targets,embedding_matrix,dropout = 0.5): def __init__(self,version,embedding_dim, hidden_dim, vocab_size, n_targets,embedding_matrix,dropout = 0.5):
@ -24,7 +24,7 @@ __Args__
2. target - a numpy array of shape [1xM] and dtype int, where M is the length of the target and each entry is the corresponding index of the word in the `embedding_matrix 2. target - a numpy array of shape [1xM] and dtype int, where M is the length of the target and each entry is the corresponding index of the word in the `embedding_matrix
__Returns__ __Returns__
1. target_scores - a torch float Tensor of shape [1xn_targets], where N is the number of dataset classes. This is the log likelihood probabilities of all the classes 1. target_scores - a torch float Tensor of shape [1xn_targets], where N is the number of dataset classes. This is the log likelihood probabilities of all the classes
@ -59,10 +59,7 @@ if dataset == 'EC':
topic = "climate change is a real concern" topic = "climate change is a real concern"
elif dataset == 'FM' : elif dataset == 'FM' :
topic = "feminist movement" topic = "feminist movement"
elif dataset == 'VCA':
topic = "vaccines cause autism"
elif dataset == 'VTI':
topic = "vaccines treat influenza"
``` ```
2. version is one of ["lstm","tan-","tan+"] 2. version is one of ["lstm","tan-","tan+"]