|
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