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

added more detection categoies

parent 7b7f287c
Branches
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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment