| 307 | |
| 308 | @pytest.mark.skip(reason="Test is flakey when run with others") |
| 309 | def test_simple_train(): |
| 310 | nlp = Language() |
| 311 | textcat = nlp.add_pipe("textcat") |
| 312 | textcat.add_label("answer") |
| 313 | nlp.initialize() |
| 314 | for i in range(5): |
| 315 | for text, answer in [ |
| 316 | ("aaaa", 1.0), |
| 317 | ("bbbb", 0), |
| 318 | ("aa", 1.0), |
| 319 | ("bbbbbbbbb", 0.0), |
| 320 | ("aaaaaa", 1), |
| 321 | ]: |
| 322 | nlp.update((text, {"cats": {"answer": answer}})) |
| 323 | doc = nlp("aaa") |
| 324 | assert "answer" in doc.cats |
| 325 | assert doc.cats["answer"] >= 0.5 |
| 326 | |
| 327 | |
| 328 | @pytest.mark.skip(reason="Test is flakey when run with others") |