Files
MoFin/venv/lib/python3.12/site-packages/nltk/test/unit/test_classify.py
T
知微 fa45d8aa5f fix: 小果地址统一node122(兼容LAN+EasyTier)
- health_checklist.json: 192.168.1.122→node122
- ocr_client.py: docstring IP→node122
- docs/market-data-requirements.md: IP→node122
- 所有API调用通过ProxyHandler({})绕过系统代理
  Privoxy对node122:18003返回500,直连正常
2026-06-30 02:56:35 +08:00

51 lines
1.3 KiB
Python

"""
Unit tests for nltk.classify. See also: nltk/test/classify.doctest
"""
import pytest
from nltk import classify
TRAIN = [
(dict(a=1, b=1, c=1), "y"),
(dict(a=1, b=1, c=1), "x"),
(dict(a=1, b=1, c=0), "y"),
(dict(a=0, b=1, c=1), "x"),
(dict(a=0, b=1, c=1), "y"),
(dict(a=0, b=0, c=1), "y"),
(dict(a=0, b=1, c=0), "x"),
(dict(a=0, b=0, c=0), "x"),
(dict(a=0, b=1, c=1), "y"),
]
TEST = [
(dict(a=1, b=0, c=1)), # unseen
(dict(a=1, b=0, c=0)), # unseen
(dict(a=0, b=1, c=1)), # seen 3 times, labels=y,y,x
(dict(a=0, b=1, c=0)), # seen 1 time, label=x
]
RESULTS = [(0.16, 0.84), (0.46, 0.54), (0.41, 0.59), (0.76, 0.24)]
def assert_classifier_correct(algorithm):
try:
classifier = classify.MaxentClassifier.train(
TRAIN, algorithm, trace=0, max_iter=1000
)
except (LookupError, AttributeError) as e:
pytest.skip(str(e))
for (px, py), featureset in zip(RESULTS, TEST):
pdist = classifier.prob_classify(featureset)
assert abs(pdist.prob("x") - px) < 1e-2, (pdist.prob("x"), px)
assert abs(pdist.prob("y") - py) < 1e-2, (pdist.prob("y"), py)
def test_megam():
assert_classifier_correct("MEGAM")
def test_tadm():
assert_classifier_correct("TADM")