diff --git a/document.go b/document.go index 6746861..aba0429 100644 --- a/document.go +++ b/document.go @@ -113,7 +113,7 @@ func NewDocument(text string, opts ...DocOpt) (*Document, error) { } if doc.Model == nil { - doc.Model = defaultModel(base.Tag, base.Extract) + doc.Model = DefaultModel(base.Tag, base.Extract) } if base.Segment { diff --git a/model.go b/model.go index 0e59b54..5e645f1 100644 --- a/model.go +++ b/model.go @@ -52,7 +52,7 @@ type EntityContext struct { // ModelFromData creates a new Model from user-provided training data. func ModelFromData(name string, sources ...DataSource) *Model { - model := defaultModel(true, true) + model := DefaultModel(true, true) model.Name = name for _, source := range sources { source(model) @@ -155,7 +155,7 @@ func loadClassifier(filesys fs.FS) *entityExtracter { return newTrainedEntityExtracter(model) } -func defaultModel(tagging, classifying bool) *Model { +func DefaultModel(tagging, classifying bool) *Model { var tagger *perceptronTagger var classifier *entityExtracter