Search in sources :

Example 26 with Config

use of com.typesafe.config.Config in project aerosolve by airbnb.

the class NormalizeFloatTransformTest method testTransform.

@Test
public void testTransform() {
    Config config = ConfigFactory.parseString(makeConfig());
    Transform transform = TransformFactory.createTransform(config, "test_norm");
    FeatureVector featureVector = TransformTestingHelper.makeFeatureVector();
    transform.doTransform(featureVector);
    Map<String, Set<String>> stringFeatures = featureVector.getStringFeatures();
    assertTrue(stringFeatures.size() == 1);
    Map<String, Double> feat1 = featureVector.getFloatFeatures().get("loc");
    assertEquals(3, feat1.size());
    double scale = 1.0 / Math.sqrt(37.7 * 37.7 + 40.0 * 40.0 + 20.0 * 20.0);
    assertEquals(scale * 37.7, feat1.get("lat"), 0.1);
    assertEquals(scale * 40.0, feat1.get("long"), 0.1);
    assertEquals(scale * -20.0, feat1.get("z"), 0.1);
}
Also used : FeatureVector(com.airbnb.aerosolve.core.FeatureVector) Config(com.typesafe.config.Config) Test(org.junit.Test)

Example 27 with Config

use of com.typesafe.config.Config in project aerosolve by airbnb.

the class NormalizeUtf8TransformTest method testEmptyFeatureVector.

@Test
public void testEmptyFeatureVector() {
    Config config = ConfigFactory.parseString(makeConfigWithoutNormalizationFormAndOutput());
    Transform transform = TransformFactory.createTransform(config, "test_normalize_utf_8");
    FeatureVector featureVector = new FeatureVector();
    transform.doTransform(featureVector);
    assertTrue(featureVector.getStringFeatures() == null);
}
Also used : FeatureVector(com.airbnb.aerosolve.core.FeatureVector) Config(com.typesafe.config.Config) Test(org.junit.Test)

Example 28 with Config

use of com.typesafe.config.Config in project aerosolve by airbnb.

the class NormalizeUtf8TransformTest method testTransformNfdNormalizationForm.

@Test
public void testTransformNfdNormalizationForm() {
    Config config = ConfigFactory.parseString(makeConfigWithNormalizationForm("NFD"));
    Transform transform = TransformFactory.createTransform(config, "test_normalize_utf_8");
    FeatureVector featureVector = makeFeatureVector();
    Map<String, Set<String>> stringFeatures = featureVector.getStringFeatures();
    transform.doTransform(featureVector);
    assertNotNull(stringFeatures);
    assertEquals(2, stringFeatures.size());
    Set<String> output = stringFeatures.get("bar");
    assertNotNull(output);
    assertEquals(1, output.size());
    assertTrue(output.contains("Funky string: ϓϔẛ"));
    assertTrue(Normalizer.isNormalized("Funky string: ϓϔẛ", Normalizer.Form.NFD));
}
Also used : FeatureVector(com.airbnb.aerosolve.core.FeatureVector) Set(java.util.Set) HashSet(java.util.HashSet) Config(com.typesafe.config.Config) Test(org.junit.Test)

Example 29 with Config

use of com.typesafe.config.Config in project aerosolve by airbnb.

the class NormalizeUtf8TransformTest method testTransformNfkdNormalizationForm.

@Test
public void testTransformNfkdNormalizationForm() {
    Config config = ConfigFactory.parseString(makeConfigWithNormalizationForm("NFKD"));
    Transform transform = TransformFactory.createTransform(config, "test_normalize_utf_8");
    FeatureVector featureVector = makeFeatureVector();
    Map<String, Set<String>> stringFeatures = featureVector.getStringFeatures();
    transform.doTransform(featureVector);
    assertNotNull(stringFeatures);
    assertEquals(2, stringFeatures.size());
    Set<String> output = stringFeatures.get("bar");
    assertNotNull(output);
    assertEquals(1, output.size());
    assertTrue(output.contains("Funky string: ΎΫṡ"));
    assertTrue(Normalizer.isNormalized("Funky string: ΎΫṡ", Normalizer.Form.NFKD));
}
Also used : FeatureVector(com.airbnb.aerosolve.core.FeatureVector) Set(java.util.Set) HashSet(java.util.HashSet) Config(com.typesafe.config.Config) Test(org.junit.Test)

Example 30 with Config

use of com.typesafe.config.Config in project aerosolve by airbnb.

the class CustomLinearLogQuantizeTransformTest method testTransform.

@Test
public void testTransform() {
    Config config = ConfigFactory.parseString(makeConfig());
    Transform transform = TransformFactory.createTransform(config, "test_quantize");
    FeatureVector featureVector = makeFeatureVector();
    transform.doTransform(featureVector);
    Map<String, Set<String>> stringFeatures = featureVector.getStringFeatures();
    assertTrue(stringFeatures.size() == 2);
    Set<String> out = stringFeatures.get("loc_quantized");
    log.info("quantize output");
    for (String string : out) {
        log.info(string);
    }
    assertTrue(out.size() == 10);
    log.info("quantize output");
    for (String string : out) {
        log.info(string);
    }
    assertTrue(out.contains("a=0.0"));
    assertTrue(out.contains("b=0.125"));
    assertTrue(out.contains("c=1.0"));
    assertTrue(out.contains("d=5.0"));
    assertTrue(out.contains("e=16.0"));
    assertTrue(out.contains("f=90.0"));
    assertTrue(out.contains("g=350.0"));
    assertTrue(out.contains("h=10000.0"));
    assertTrue(out.contains("i=-1.0"));
    assertTrue(out.contains("j=-22.0"));
}
Also used : FeatureVector(com.airbnb.aerosolve.core.FeatureVector) Config(com.typesafe.config.Config) Test(org.junit.Test)

Aggregations

Config (com.typesafe.config.Config)138 Test (org.junit.Test)120 FeatureVector (com.airbnb.aerosolve.core.FeatureVector)117 Set (java.util.Set)35 HashSet (java.util.HashSet)29 HashMap (java.util.HashMap)19 Map (java.util.Map)19 ActorSystem (akka.actor.ActorSystem)2 ConfigObject (com.typesafe.config.ConfigObject)2 ConfigValue (com.typesafe.config.ConfigValue)2 IOException (java.io.IOException)2 URI (java.net.URI)2 ArrayList (java.util.ArrayList)2 List (java.util.List)2 DrillConfig (org.apache.drill.common.config.DrillConfig)2 BeforeClass (org.junit.BeforeClass)2 Example (com.airbnb.aerosolve.core.Example)1 Stopwatch (com.google.common.base.Stopwatch)1 ServerConfig (com.nike.riposte.server.config.ServerConfig)1 ContainerConfig (com.spotify.docker.client.messages.ContainerConfig)1