PSOpenAI

3.7.0

Public/Get-CosineSimilarity.ps1

class VectorUtils {
    static [double] Dot ([double[]]$Vector1, [double[]]$Vector2) {
        if ($Vector1.Length -ne $Vector2.Length) {
            throw [System.ArgumentException]::new('All vectors must have the same dimensionality.')
        }

        [double]$dot = .0d
        for ($i = 0; $i -lt $Vector1.Count; $i++) {
            $dot += $Vector1[$i] * $Vector2[$i]
        }
        return $dot
    }

    static [double] Norm ([double[]]$Vector1) {
        return [System.Math]::Sqrt([VectorUtils]::Dot($Vector1, $Vector1))
    }

    static [double] CosineSimilarity ([double[]]$Vector1, [double[]]$Vector2) {
        return [VectorUtils]::Dot($Vector1, $Vector2) / [VectorUtils]::Norm($Vector1) * [VectorUtils]::Norm($Vector2)
    }

    static [double] CosineDistance ([double[]]$Vector1, [double[]]$Vector2) {
        return 1.0 - [VectorUtils]::CosineSimilarity($Vector1, $Vector2)
    }
}

function Get-CosineSimilarity {
    [CmdletBinding()]
    [OutputType([double])]
    param (
        [Parameter(Mandatory, Position = 0)]
        [double[]]$Vector1,

        [Parameter(Mandatory, Position = 1)]
        [double[]]$Vector2
    )
    try {
        [VectorUtils]::CosineSimilarity($Vector1, $Vector2)
    }
    catch {
        Write-Error -Exception $_.Exception
    }
}