Spaces:
Running
Running
File size: 1,820 Bytes
b110593 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
// _ _
// __ _____ __ ___ ___ __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
// \ V V / __/ (_| |\ V /| | (_| | || __/
// \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
// Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
// CONTACT: [email protected]
//
package nearText
import (
"fmt"
)
type movements struct{}
func newMovements() *movements {
return &movements{}
}
// MoveTo moves one vector toward another
func (v *movements) MoveTo(source []float32, target []float32, weight float32,
) ([]float32, error) {
multiplier := float32(0.5)
if len(source) != len(target) {
return nil, fmt.Errorf("movement: vector lengths don't match: got %d and %d",
len(source), len(target))
}
if weight < 0 || weight > 1 {
return nil, fmt.Errorf("movement: force must be between 0 and 1: got %f",
weight)
}
out := make([]float32, len(source))
for i, sourceItem := range source {
out[i] = sourceItem*(1-weight*multiplier) + target[i]*(weight*multiplier)
}
return out, nil
}
// MoveAwayFrom moves one vector away from another
func (v *movements) MoveAwayFrom(source []float32, target []float32, weight float32,
) ([]float32, error) {
multiplier := float32(0.5) // so the movement is fair in comparison with moveTo
if len(source) != len(target) {
return nil, fmt.Errorf("movement (moveAwayFrom): vector lengths don't match: "+
"got %d and %d", len(source), len(target))
}
if weight < 0 {
return nil, fmt.Errorf("movement (moveAwayFrom): force must be 0 or positive: "+
"got %f", weight)
}
out := make([]float32, len(source))
for i, sourceItem := range source {
out[i] = sourceItem + weight*multiplier*(sourceItem-target[i])
}
return out, nil
}
|