()
| 154 | |
| 155 | |
| 156 | def main(): |
| 157 | Xtrain, Ytrain, Xtest, Ytest, word2idx = get_data(split_sequences=True) |
| 158 | V = len(word2idx) + 1 |
| 159 | K = len(set(flatten(Ytrain)) | set(flatten(Ytest))) |
| 160 | rnn = RNN(10, [10], V, K) |
| 161 | rnn.fit(Xtrain, Ytrain) |
| 162 | print("train score:", rnn.score(Xtrain, Ytrain)) |
| 163 | print("test score:", rnn.score(Xtest, Ytest)) |
| 164 | print("train f1:", rnn.f1_score(Xtrain, Ytrain)) |
| 165 | print("test f1:", rnn.f1_score(Xtest, Ytest)) |
| 166 | |
| 167 | |
| 168 | if __name__ == '__main__': |