Package com.linkedin.venice.compute.protocol.request

Class CosineSimilarity

java.lang.Object
org.apache.avro.specific.SpecificRecordBase
com.linkedin.venice.compute.protocol.request.CosineSimilarity
All Implemented Interfaces:
Externalizable, Serializable, Comparable<org.apache.avro.specific.SpecificRecord>, org.apache.avro.generic.GenericContainer, org.apache.avro.generic.GenericRecord, org.apache.avro.generic.IndexedRecord, org.apache.avro.specific.SpecificRecord
public class CosineSimilarity extends org.apache.avro.specific.SpecificRecordBase implements org.apache.avro.specific.SpecificRecord
See Also:

  • Field Details

    • SCHEMA$

      public static final org.apache.avro.Schema SCHEMA$

    • field

      public CharSequence field
      The field in the original value record, which will used to execute cosine-similarity calculation

    • cosSimilarityParam

      public List<Float> cosSimilarityParam
      The passed feature vector, which will be used to execute cosine-similarity calculation against the field in the original value record

    • resultFieldName

      public CharSequence resultFieldName
      The field name used to store the calculated result

  • Constructor Details

    • CosineSimilarity

      public CosineSimilarity()
      Default constructor. Note that this does not initialize fields to their default values from the schema. If that is desired then one should use newBuilder().

    • CosineSimilarity

      public CosineSimilarity(CharSequence field, List<Float> cosSimilarityParam, CharSequence resultFieldName)
      All-args constructor.
      Parameters:
      field - The field in the original value record, which will used to execute cosine-similarity calculation
      cosSimilarityParam - The passed feature vector, which will be used to execute cosine-similarity calculation against the field in the original value record
      resultFieldName - The field name used to store the calculated result

  • Method Details

    • getClassSchema

      public static org.apache.avro.Schema getClassSchema()

    • getSpecificData

      public org.apache.avro.specific.SpecificData getSpecificData()
      Overrides:
      getSpecificData in class org.apache.avro.specific.SpecificRecordBase

    • getSchema

      public org.apache.avro.Schema getSchema()
      Specified by:
      getSchema in interface org.apache.avro.generic.GenericContainer
      Specified by:
      getSchema in class org.apache.avro.specific.SpecificRecordBase

    • get

      public Object get(int field$)
      Specified by:
      get in interface org.apache.avro.generic.IndexedRecord
      Specified by:
      get in class org.apache.avro.specific.SpecificRecordBase

    • put

      public void put(int field$, Object value$)
      Specified by:
      put in interface org.apache.avro.generic.IndexedRecord
      Specified by:
      put in class org.apache.avro.specific.SpecificRecordBase

    • getField

      public CharSequence getField()
      Gets the value of the 'field' field.
      Returns:
      The field in the original value record, which will used to execute cosine-similarity calculation

    • setField

      public void setField(CharSequence value)
      Sets the value of the 'field' field. The field in the original value record, which will used to execute cosine-similarity calculation
      Parameters:
      value - the value to set.

    • getCosSimilarityParam

      public List<Float> getCosSimilarityParam()
      Gets the value of the 'cosSimilarityParam' field.
      Returns:
      The passed feature vector, which will be used to execute cosine-similarity calculation against the field in the original value record

    • setCosSimilarityParam

      public void setCosSimilarityParam(List<Float> value)
      Sets the value of the 'cosSimilarityParam' field. The passed feature vector, which will be used to execute cosine-similarity calculation against the field in the original value record
      Parameters:
      value - the value to set.

    • getResultFieldName

      public CharSequence getResultFieldName()
      Gets the value of the 'resultFieldName' field.
      Returns:
      The field name used to store the calculated result

    • setResultFieldName

      public void setResultFieldName(CharSequence value)
      Sets the value of the 'resultFieldName' field. The field name used to store the calculated result
      Parameters:
      value - the value to set.

    • writeExternal

      public void writeExternal(ObjectOutput out) throws IOException
      Specified by:
      writeExternal in interface Externalizable
      Overrides:
      writeExternal in class org.apache.avro.specific.SpecificRecordBase
      Throws:
      IOException

    • readExternal

      public void readExternal(ObjectInput in) throws IOException
      Specified by:
      readExternal in interface Externalizable
      Overrides:
      readExternal in class org.apache.avro.specific.SpecificRecordBase
      Throws:
      IOException