From 36cb680e85301e25d2512c8bb254084f22625700 Mon Sep 17 00:00:00 2001 From: Alex Bilbie Date: Tue, 17 May 2022 12:02:33 +0100 Subject: [PATCH] Expose defaultModel function --- document.go | 2 +- model.go | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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