private SimpleMatrix GetMentionFeatures(Mention m, Document document, IDictionary <int, IList <Mention> > mentionsByHeadIndex) { return(NeuralUtils.Concatenate(NeuralUtils.OneHot((int)(m.mentionType), 4), EncodeDistance(m.endIndex - m.startIndex - 1), new SimpleMatrix(new double[][] { new double[] { m.mentionNum / (double)document.predictedMentionsByID.Count }, new double [] { mentionsByHeadIndex[m.headIndex].Stream().AnyMatch(null) ? 1 : 0 } }))); }
private SimpleMatrix EncodeGenre(Document document) { return(conll ? NeuralUtils.OneHot(genres[document.docInfo["DOC_ID"].Split("/")[0]], genres.Count) : new SimpleMatrix(1, 1)); }