explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-28 16:04:55 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update nlp.config (workaround).

Browse Source
This commit is contained in:
Raphael Mitsch 2022-09-02 16:25:57 +02:00
parent 24b69a1be8
commit 73432c6bfb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/cli/find_threshold.py
View File

@ -1,5 +1,4 @@
import functools
from functools import partial
import operator
from pathlib import Path
import logging
@ -7,6 +6,7 @@ from typing import Optional, Tuple, Any, Dict, List
import numpy
import wasabi.tables
from confection import Config
from ..pipeline import TrainablePipe, Pipe
from ..errors import Errors
@ -121,7 +121,6 @@ def find_threshold(
raise TypeError(Errors.E1044)
if not hasattr(pipe, "scorer"):
raise AttributeError(Errors.E1045)
setattr(pipe, "scorer", partial(pipe.scorer.func, beta=beta))
if not silent:
wasabi.msg.info(
@ -150,6 +149,7 @@ def find_threshold(
scores: Dict[float, float] = {}
for threshold in numpy.linspace(0, 1, n_trials):
pipe.cfg = set_nested_item(pipe.cfg, config_keys, threshold)
nlp._pipe_configs[pipe_name] = Config(pipe.cfg)
scores[threshold] = nlp.evaluate(dev_dataset)[scores_key]
if not (
isinstance(scores[threshold], float) or isinstance(scores[threshold], int)