|
8 | 8 | import edu.stanford.nlp.neural.NeuralUtils; |
9 | 9 | import org.ejml.simple.SimpleMatrix; |
10 | 10 |
|
11 | | -// TODO: remove when ejml is upgraded |
12 | | -import java.io.IOException; |
13 | | -import java.io.ObjectInputStream; |
14 | | -import java.util.stream.Collectors; |
15 | | - |
16 | 11 | /** |
17 | 12 | * Stores the weights and implements the matrix operations used by a {@link NeuralCorefAlgorithm} |
18 | 13 | * @author Kevin Clark |
19 | 14 | */ |
20 | 15 | public class NeuralCorefModel implements Serializable { |
21 | 16 | private static final long serialVersionUID = 2139427931784505653L; |
22 | | - // TODO: restore /*final*/ when ejml is upgraded |
23 | | - private /*final*/ SimpleMatrix antecedentMatrix; |
24 | | - private /*final*/ SimpleMatrix anaphorMatrix; |
25 | | - private /*final*/ SimpleMatrix pairFeaturesMatrix; |
26 | | - private /*final*/ SimpleMatrix pairwiseFirstLayerBias; |
27 | | - private /*final*/ List<SimpleMatrix> anaphoricityModel; |
28 | | - private /*final*/ List<SimpleMatrix> pairwiseModel; |
29 | | - private /*final*/ Embedding wordEmbeddings; |
| 17 | + |
| 18 | + private SimpleMatrix antecedentMatrix; |
| 19 | + private SimpleMatrix anaphorMatrix; |
| 20 | + private SimpleMatrix pairFeaturesMatrix; |
| 21 | + private SimpleMatrix pairwiseFirstLayerBias; |
| 22 | + private List<SimpleMatrix> anaphoricityModel; |
| 23 | + private List<SimpleMatrix> pairwiseModel; |
| 24 | + private Embedding wordEmbeddings; |
30 | 25 |
|
31 | 26 | public NeuralCorefModel(SimpleMatrix antecedentMatrix, SimpleMatrix anaphorMatrix, |
32 | 27 | SimpleMatrix pairFeaturesMatrix, SimpleMatrix pairwiseFirstLayerBias, |
|
0 commit comments