/** * Sanitize the entities by removing punctuation and other attributes. * * @param entities A collection of {@link Entity} objects. * @return A collection of sanitized {@link Entity} objects. This collection * will be equal in size to the input collection. */ @Override public Set<Entity> sanitizeEntities(Set<Entity> entities) { for(Entity entity : entities) { if(entity.getText().endsWith(",") || entity.getText().endsWith(".")) { entity.setText(entity.getText().substring(0, entity.getText().length() - 1)); } // Replace all punctuation. entity.setText(entity.getText().replaceAll("\\p{P}", "")); } return entities; }
/** * Create a new entity from an existing entity. * @param entity An existing entity. */ public Entity(Entity entity) { setConfidence(entity.getConfidence()); setMetadata(entity.getMetadata()); setType(entity.getType()); setLanguageCode(entity.getLanguageCode()); setSpan(entity.getSpan()); setText(entity.getText()); setUri(entity.getUri()); }