Skip to content

Conversation

@Pringled
Copy link
Member

This PR adds multilabel classification to the training module. If labels are passed as a list[list[str]], the model is automatically trained as a multilabel classification model.

@Pringled Pringled requested a review from stephantul February 14, 2025 11:20
@codecov
Copy link

codecov bot commented Feb 14, 2025

Codecov Report

Attention: Patch coverage is 98.01980% with 2 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
model2vec/train/classifier.py 96.49% 2 Missing ⚠️
Files with missing lines Coverage Δ
model2vec/inference/model.py 90.58% <100.00%> (+2.01%) ⬆️
tests/conftest.py 100.00% <100.00%> (ø)
tests/test_inference.py 100.00% <100.00%> (ø)
tests/test_trainable.py 100.00% <100.00%> (ø)
model2vec/train/classifier.py 97.31% <96.49%> (-0.10%) ⬇️
@Pringled Pringled requested a review from stephantul February 14, 2025 14:29
@Pringled Pringled marked this pull request as ready for review February 15, 2025 13:52
@Pringled Pringled merged commit 2d51516 into main Feb 16, 2025
6 checks passed
@Pringled Pringled deleted the add-multilabel-classification branch February 16, 2025 13:22
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

3 participants