From 60ecd977e7599f382a25aefeccaab5fe76703dfe Mon Sep 17 00:00:00 2001 From: Ian Roberts Date: Mon, 14 Sep 2020 11:47:25 +0100 Subject: [PATCH] Use a supported model type by default, but allow it to be overridden by a parameter from the PythonPR --- gate_stance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gate_stance.py b/gate_stance.py index 25cc2fa..3a16356 100644 --- a/gate_stance.py +++ b/gate_stance.py @@ -32,7 +32,7 @@ def run(doc, **kwargs): global classifier if classifier is None: - classifier = StanceClassifier("lr") + classifier = StanceClassifier(kwargs.get("model", "ens")) text = doc.text