From f99a6a2eba4f0f131b257118e14378d97dc3f358 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C5=82=20Pogoda?= <mipo57@e-science.pl>
Date: Fri, 31 Mar 2023 09:59:58 +0200
Subject: [PATCH] added more detection categoies
---
src/detections/detection.py | 35 +++++++++++++++++++++++++++++++++++
1 file changed, 35 insertions(+)
diff --git a/src/detections/detection.py b/src/detections/detection.py
index 96e60c7..f3aa0b8 100644
--- a/src/detections/detection.py
+++ b/src/detections/detection.py
@@ -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"
--
GitLab