Skip to content

Commit

Permalink
Add default model for Italian text categorization
Browse files Browse the repository at this point in the history
  • Loading branch information
matteo-grella committed Sep 23, 2022
1 parent 178fa73 commit 3c30cb6
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions pkg/tasks/textclassification/textclassification.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,13 @@ package textclassification

import "context"

const (
// DefaultModelForItalianNewsClassification is a model fine-tuned for news headlines classification in Italian.
// It predicts the top-level category of the IPTC subject taxonomy: https://cv.iptc.org/newscodes/subjectcode
// Model card: https://huggingface.co/nlpodyssey/bert-italian-uncased-iptc-headlines
DefaultModelForItalianNewsClassification = "nlpodyssey/bert-italian-uncased-iptc-headlines"
)

// Interface defines the main functions for text classification task.
type Interface interface {
// Classify returns the classification of the given example.
Expand Down

0 comments on commit 3c30cb6

Please sign in to comment.