38 lines
759 B
Markdown
38 lines
759 B
Markdown
|
```go title="Go"
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"github.com/kreuzberg-dev/kreuzberg/packages/go/v5"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
maxKeywords := int32(10)
|
||
|
|
minScore := 0.3
|
||
|
|
|
||
|
|
config := &kreuzberg.ExtractionConfig{
|
||
|
|
Keywords: &kreuzberg.KeywordConfig{
|
||
|
|
Algorithm: kreuzberg.KeywordAlgorithm_YAKE,
|
||
|
|
MaxKeywords: &maxKeywords,
|
||
|
|
MinScore: &minScore,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
result, err := kreuzberg.ExtractFileSync("research_paper.pdf", config)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("extraction failed: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if keywords, ok := result.Metadata["keywords"]; ok {
|
||
|
|
keywordList := keywords.([]map[string]interface{})
|
||
|
|
for _, kw := range keywordList {
|
||
|
|
text := kw["text"].(string)
|
||
|
|
score := kw["score"].(float64)
|
||
|
|
fmt.Printf("%s: %.3f\n", text, score)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
```
|