90 lines
1.9 KiB
Go
90 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"kreuzberg"
|
|
)
|
|
|
|
// Example 1: Basic YAKE configuration
|
|
// Uses YAKE algorithm with default parameters and English stopword filtering
|
|
func basicYake() error {
|
|
config := &kreuzberg.ExtractionConfig{
|
|
Keywords: &kreuzberg.KeywordConfig{
|
|
Algorithm: "yake",
|
|
MaxKeywords: 10,
|
|
MinScore: 0.0,
|
|
NgramRange: [2]int{1, 3},
|
|
Language: "en",
|
|
YakeParams: nil,
|
|
RakeParams: nil,
|
|
},
|
|
}
|
|
|
|
result, err := kreuzberg.ExtractFileSync("document.pdf", config)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("Keywords: %v\n", result.Keywords)
|
|
return nil
|
|
}
|
|
|
|
// Example 2: Advanced YAKE with custom parameters
|
|
// Fine-tunes YAKE with custom window size for co-occurrence analysis
|
|
func advancedYake() error {
|
|
config := &kreuzberg.ExtractionConfig{
|
|
Keywords: &kreuzberg.KeywordConfig{
|
|
Algorithm: "yake",
|
|
MaxKeywords: 15,
|
|
MinScore: 0.1,
|
|
NgramRange: [2]int{1, 2},
|
|
Language: "en",
|
|
YakeParams: &kreuzberg.YakeParams{
|
|
WindowSize: 1,
|
|
},
|
|
RakeParams: nil,
|
|
},
|
|
}
|
|
|
|
result, err := kreuzberg.ExtractFileSync("document.pdf", config)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("Keywords: %v\n", result.Keywords)
|
|
return nil
|
|
}
|
|
|
|
// Example 3: RAKE configuration
|
|
// Uses RAKE algorithm for rapid keyword extraction with phrase constraints
|
|
func rakeConfig() error {
|
|
config := &kreuzberg.ExtractionConfig{
|
|
Keywords: &kreuzberg.KeywordConfig{
|
|
Algorithm: "rake",
|
|
MaxKeywords: 10,
|
|
MinScore: 5.0,
|
|
NgramRange: [2]int{1, 3},
|
|
Language: "en",
|
|
YakeParams: nil,
|
|
RakeParams: &kreuzberg.RakeParams{
|
|
MinWordLength: 1,
|
|
MaxWordsPerPhrase: 3,
|
|
},
|
|
},
|
|
}
|
|
|
|
result, err := kreuzberg.ExtractFileSync("document.pdf", config)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Printf("Keywords: %v\n", result.Keywords)
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
if err := basicYake(); err != nil {
|
|
fmt.Println("Error:", err)
|
|
}
|
|
}
|