Awesome
Snowball
A Go (golang) implementation of the Snowball stemmer for natural language processing.
Status | |
---|---|
Latest release | v0.10.0 (2024-08-13) |
Latest build status | |
Languages available | English, Spanish (español), French (le français), Russian (ру́сский язы́к), Swedish (svenska), Norwegian (norsk), Hungarian (magyar) |
License | MIT |
Usage
Here is a minimal Go program that uses this package in order to stem a single word.
package main
import (
"fmt"
"github.com/kljensen/snowball"
)
func main(){
stemmed, err := snowball.Stem("Accumulations", "english", true)
if err == nil{
fmt.Println(stemmed) // Prints "accumul"
}
}
Organization & Implementation
The code is organized as follows:
- The top-level
snowball
package has a single exported functionsnowball.Stem
, which is defined insnowball/snowball.go
. - The stemmer for each language is defined in a "sub-package", e.g
snowball/spanish
. - Each language exports a
Stem
function: e.g.spanish.Stem
, which is defined insnowball/spanish/stem.go
. - Code that is common to multiple languages may go in a separate package,
e.g. the small
romance
package.
Some notes about the implementation:
- In order to ensure the code is easily extended to non-English languages,
I avoided using bytes and byte arrays, and instead perform all operations
on runes. See
snowball/snowballword/snowballword.go
and theSnowballWord
struct. - In order to avoid casting strings into slices of runes numerous times,
this implementation uses a single slice of runes stored in the
SnowballWord
struct for each word that needs to be stemmed. - In spite of the foregoing, readability requires that some strings be kept around and repeatedly cast into slices of runes. For example, in the Spanish stemmer, one step requires removing suffixes with accute accents such as "ución", "logía", and "logías". If I were to hard-code those suffices as slices of runes, the code would be substantially less readable.
- Instead of carrying around the word regions R1, R2, & RV as separate strings
(or slices or runes, or whatever), we carry around the index where each of
these regions begins. These are stored as
R1start
,R2start
, &RVstart
on theSnowballWord
struct. I believe this is a relatively efficient way of storing R1 and R2. - The code does not use any maps or regular expressions 1) for kicks, and 2) because I thought they'd negatively impact the performance. (But, mostly for #1; I realize #2 is silly.)
- I end up refactoring the
snowballword
package a bit every time I implement a new language. - Clearly, the Go implentation of these stemmers is verbose relative to the Snowball language. However, it is much better than the Java version and others.
Testing
To run the tests, do go test ./...
in the top-level directory.
Future work
I'd like to implement the Snowball stemmer in more languages. If you can help, I would greatly appreciate it: please fork the project and send a pull request!
(Also, if you are interested in creating a larger NLP project for Go, please get in touch.)
Related work
I know of a few other stemmers availble in Go:
- stemmer by Dmitry Chestnykh. His project also implements the Snowball (Porter2) English stemmer as well as the Snowball German stemmer.
- porter-stemmer - an implementation of the original Porter stemming algorithm.
- go-stem by Alex Gonopolskiy. Also the original Porter algorithm.
- paicehusk by Aaron Groves. This package implements the Paice/Husk stemmer.
- golibstemmer by Richard Johnson. This provides Go bindings for the libstemmer C library.
- snowball by Miki Tebeka. Also, I believe, Go bindings for the C library.
Contributors
- Kyle Jensen (kljensen@gmail.com, @DataKyle)
- Shawn Smith
- Herman Schaaf
- Anton Södergren
- Eivind Moland
- Tamás Gulácsi
- @clipperhouse
- Your name should be here!
License (MIT)
Copyright (c) the Contributors (see above)
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.