Skip to content
Snippets Groups Projects
Commit f99a6a2e authored by Michał Pogoda's avatar Michał Pogoda
Browse files

added more detection categoies

parent 7b7f287c
No related branches found
No related tags found
3 merge requests!10Anonimizer v2,!9Fix infancy erorrs based on Magdalena's report,!7Better coverage
......@@ -32,6 +32,35 @@ class SurnameDetection(MorphosyntacticInfoMixin, Detection):
def __init__(self, morpho_tag: Optional[str] = None) -> None:
super().__init__(morpho_tag=morpho_tag)
class LocationDetection(MorphosyntacticInfoMixin, Detection):
TYPE_NAME = "location"
def __init__(self, morpho_tag: Optional[str] = None) -> None:
super().__init__(morpho_tag=morpho_tag)
class OrganizationNameDetection(MorphosyntacticInfoMixin, Detection):
TYPE_NAME = "organization_name"
def __init__(self, morpho_tag: Optional[str] = None) -> None:
super().__init__(morpho_tag=morpho_tag)
def ProperNameDetection(MorphosyntacticInfoMixin, Detection):
TYPE_NAME = "proper_name"
def __init__(self, morpho_tag: Optional[str] = None) -> None:
super().__init__(morpho_tag=morpho_tag)
class TitleDetection(MorphosyntacticInfoMixin, Detection):
TYPE_NAME = "title"
def __init__(self, morpho_tag: Optional[str] = None) -> None:
super().__init__(morpho_tag=morpho_tag)
class HydronymDetection(MorphosyntacticInfoMixin, Detection):
TYPE_NAME = "hydronym"
def __init__(self, morpho_tag: Optional[str] = None) -> None:
super().__init__(morpho_tag=morpho_tag)
class StreetNameDetection(MorphosyntacticInfoMixin, Detection):
TYPE_NAME = "street_name"
......@@ -47,6 +76,7 @@ class CityDetection(MorphosyntacticInfoMixin, Detection):
super().__init__(morpho_tag=morpho_tag)
class CountryDetection(MorphosyntacticInfoMixin, Detection):
TYPE_NAME = "country"
......@@ -102,6 +132,11 @@ class KRSDetection(Detection): # National Court Register
def __init__(self) -> None:
super().__init__()
class SerialNumberDetection(Detection):
TYPE_NAME = "serial_number"
def __init__(self) -> None:
super().__init__()
class OtherDetection(Detection): # Non standard entity
TYPE_NAME = "other"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment