classifier
classifier copied to clipboard
n3integration
A general purpose text classifier
classifier
A naive bayes text classifier.
Installation
go get github.com/n3integration/classifier
Usage
Classification
There are two methods of classification: io.Reader or string. To classify strings, use the TrainString or ClassifyString functions. To classify larger sources, use the Train and Classify functions that take an io.Reader as input.
import "github.com/n3integration/classifier/naive"
classifier := naive.New()
classifier.TrainString("The quick brown fox jumped over the lazy dog", "ham")
classifier.TrainString("Earn a degree online", "ham")
classifier.TrainString("Earn cash quick online", "spam")
if classification, err := classifier.ClassifyString("Earn your masters degree online"); err == nil {
fmt.Println("Classification => ", classification) // ham
} else {
fmt.Println("error: ", err)
}
Contributing
- Fork the repository
- Create a local feature branch
- Run
gofmt - Bump the
VERSIONfile using semantic versioning - Submit a pull request
License
Copyright 2022 [email protected]
Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.
n3integration