Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/buckhx/gobert
/ types & classes
Types & classes
23 in github.com/buckhx/gobert
⨍
Functions
88
◇
Types & classes
23
↓ 4 callers
TypeAlias
ID
ID is used to identify vocab items
tokenize/vocab/vocab.go:14
↓ 1 callers
Interface
ValueProvider
ValueProvider is a simple interface for tensors responses without the baggage
model/bert.go:34
Struct
Basic
Basic is a BasicTokenizer that runs Runs basic tokenize (punctuation splitting, lower casing, etc.).
tokenize/basic.go:12
Struct
Bert
Bert is a model that translates features to values from an exported model. It processes as follows: Pipeline: text -> FeatureFactory -> TensorFunc ->
model/bert.go:40
FuncType
BertOption
BertOption configures a BERT model
model/option.go:9
Struct
Dict
Dict is a container for tokens NOTE: python uses an OrderedDict, unsure of implications
tokenize/vocab/vocab.go:23
Interface
Estimator
Estimator matches the tf, p
model/estimator/estimator.go:17
Struct
Feature
Feature is an input feature for a BERT model. Maps to extract_features.InputFeature in ref-impl
tokenize/feature.go:17
Struct
FeatureFactory
FeatureFactory will create features with the supplied tokenizer and sequence length
tokenize/feature.go:38
FuncType
FeatureTensorFunc
FeatureTensorFunc translates features to tensors
model/bert.go:31
Struct
Full
Full is a FullTokenizer which comprises of a Basic & Wordpiece tokenizer
tokenize/full.go:6
FuncType
InputFunc
InputFunc matches feeds in sessions
model/estimator/estimator.go:11
FuncType
ModelFunc
ModelFunc the returned params match fetches & targets from the API
model/estimator/estimator.go:14
FuncType
Option
Option alter the behavior of the tokenizer TODO add tests for these behavior changes
tokenize/tokenizer.go:36
Interface
Predictor
Predictor creates tensors for prediction
model/estimator/estimator.go:27
Interface
Provider
Provider is an interface for exposing a vocab
tokenize/vocab/vocab.go:9
FuncType
TensorInputFunc
TensorInputFunc maps tensors to an estimator.InputFunc in the Predict pipeline
model/bert.go:28
Interface
Tokenizer
Tokenizer is an interface for chunking a string into it's tokens as per the BERT implematation
tokenize/tokenizer.go:11
Interface
VocabTokenizer
VocabTokenizer comprises of a Tokenizer and VocabProvider
tokenize/tokenizer.go:16
Struct
Wordpiece
Wordpiece is a tokenizer that breaks tokens into subword units based on a supplied vocabulary https://arxiv.org/pdf/1609.08144.pdf Section 4.1 for det
tokenize/wordpiece.go:17
Struct
engine
engine is a simple semantic search engine for demonstrating using a BERT model It is not exported because it is not meant to be used outside of demons
examples/semantic-search/engine.go:15
Struct
predictor
model/estimator/predict.go:9
Struct
rng
TODO batch from flag
examples/semantic-search/engine.go:46