Examples with FloatVector com.airbnb.aerosolve.core.util.FloatVector used on opensource projects

Example 26 with FloatVector

use of com.airbnb.aerosolve.core.util.FloatVector in project aerosolve by airbnb.

the class LowRankLinearModelTest method makeLabelWeightVector.

Map<String, FloatVector> makeLabelWeightVector() {
    Map<String, FloatVector> labelWeights = new HashMap<>();
    float[] animalFeature = { 1.0f, 0.0f, 0.0f };
    float[] colorFeature = { 0.0f, 1.0f, 0.0f };
    float[] fruitFeature = { 0.0f, 0.0f, 1.0f };
    labelWeights.put("A", new FloatVector(animalFeature));
    labelWeights.put("C", new FloatVector(colorFeature));
    labelWeights.put("F", new FloatVector(fruitFeature));
    return labelWeights;
}