private void saveRelations(String documentId, JCas jCas) { EntityRelationConverter converter = new EntityRelationConverter( getMonitor(), outputHistory, getSupport().getDocumentHistory(jCas), stopFeatures, fields); List<Document> rels = JCasUtil.select(jCas, Relation.class) .stream() .map(converter::convertRelation) .map(Document::new) .peek(d -> d.append(FIELD_DOCUMENT_ID, documentId)) .collect(toList()); if (!rels.isEmpty()) { relationsCollection.insertMany(rels); } } }
private void createERC() { final IEntityConverterFields fields = new DefaultFields(); final HashSet<String> stopFeatures = new HashSet<>(); stopFeatures.add("uima.cas.AnnotationBase:sofa"); stopFeatures.add("uk.gov.dstl.baleen.types.BaleenAnnotation:internalId"); erc = new EntityRelationConverter(monitor, stopFeatures, fields, false); }
private void saveEntities(String documentId, JCas jCas) { EntityRelationConverter converter = new EntityRelationConverter( getMonitor(), outputHistory,
new EntityRelationConverter( monitor, false, support.getDocumentHistory(jCas), stopFeatures, fields);
@Override public void doInitialize(final UimaContext aContext) throws ResourceInitializationException { super.doInitialize(aContext); final IEntityConverterFields fields = new DefaultFields(); final HashSet<String> stopFeatures = new HashSet<>(); stopFeatures.add("uima.cas.AnnotationBase:sofa"); stopFeatures.add("uk.gov.dstl.baleen.types.BaleenAnnotation:internalId"); entityRelationConverter = new EntityRelationConverter(getMonitor(), stopFeatures, fields, false); if (outputDocuments) { initialiseForDocuments(); } if (outputMentions) { initialiseForMentions(); } if (outputEntities) { initialiseForEntities(); } if (outputRelations) { initialiseForRelations(); } }
stopFeatures.add("uk.gov.dstl.baleen.types.BaleenAnnotation:internalId"); EntityRelationConverter erc = new EntityRelationConverter( getMonitor(), false,
private Optional<Vertex> traversal(JCas jCas, GraphTraversalSource traversal) { EntityRelationConverter converter = new EntityRelationConverter(monitor, options.getStopFeatures(), fields, false); loadMentions(jCas, traversal, converter); if (options.isOutputRelations()) { loadRelations(jCas, traversal, converter); } else { monitor.debug("DocumentGraph relations ommitted"); } if (options.isOutputEvents()) { loadEvents(jCas, traversal, converter); } else { monitor.debug("DocumentGraph events ommitted"); } return loadDocument(jCas, traversal); }
private Optional<Vertex> traversal(JCas jCas, GraphTraversalSource traversal) { EntityRelationConverter converter = new EntityRelationConverter(monitor, options.getStopFeatures(), fields, false); loadMentions(jCas, traversal, converter); if (options.isOutputRelations()) { loadRelations(jCas, traversal, converter); } else { monitor.debug("DocumentGraph relations ommitted"); } if (options.isOutputEvents()) { loadEvents(jCas, traversal, converter); } else { monitor.debug("DocumentGraph events ommitted"); } return loadDocument(jCas, traversal); }