Hinweis
Für den Zugriff auf diese Seite ist eine Autorisierung erforderlich. Sie können versuchen, sich anzumelden oder das Verzeichnis zu wechseln.
Für den Zugriff auf diese Seite ist eine Autorisierung erforderlich. Sie können versuchen, das Verzeichnis zu wechseln.
Gibt die Kosinus-Ähnlichkeit zwischen zwei Gleitkommavektoren zurück. Die Vektoren müssen dieselbe Dimension aufweisen.
Die entsprechende Databricks SQL-Funktion finden Sie unter vector_cosine_similarity Funktion.
Syntax
from pyspark.sql import functions as dbf
dbf.vector_cosine_similarity(left=<left>, right=<right>)
Parameter
| Parameter | Typ | Beschreibung |
|---|---|---|
left |
pyspark.sql.Column oder Spaltenname |
Erste Vektorspalte. |
right |
pyspark.sql.Column oder Spaltenname |
Zweite Vektorspalte. |
Returns
pyspark.sql.Column: Kosinus-Ähnlichkeit als Gleitkommawert.
Beispiele
from pyspark.sql import functions as dbf
from pyspark.sql.types import ArrayType, FloatType, StructType, StructField
schema = StructType([StructField('a', ArrayType(FloatType())), StructField('b', ArrayType(FloatType()))])
df = spark.createDataFrame([([1.0, 2.0, 3.0], [4.0, 5.0, 6.0])], schema)
df.select(dbf.vector_cosine_similarity('a', 'b')).first()[0]
# 0.974631...