@Test public void testNumericAttribute() { NumericAttribute attr = NumericAttribute.defaultAttr() .withName("age").withIndex(0).withMin(0.0).withMax(1.0).withStd(0.5).withSparsity(0.4); Assert.assertEquals(attr.withoutIndex(), Attribute.fromStructField(attr.toStructField())); }
@Test public void testNumericAttribute() { NumericAttribute attr = NumericAttribute.defaultAttr() .withName("age").withIndex(0).withMin(0.0).withMax(1.0).withStd(0.5).withSparsity(0.4); Assert.assertEquals(attr.withoutIndex(), Attribute.fromStructField(attr.toStructField())); }
@Test public void testNumericAttribute() { NumericAttribute attr = NumericAttribute.defaultAttr() .withName("age").withIndex(0).withMin(0.0).withMax(1.0).withStd(0.5).withSparsity(0.4); Assert.assertEquals(attr.withoutIndex(), Attribute.fromStructField(attr.toStructField())); }
@Test public void testAttributeGroup() { Attribute[] attrs = new Attribute[]{ NumericAttribute.defaultAttr(), NominalAttribute.defaultAttr(), BinaryAttribute.defaultAttr().withIndex(0), NumericAttribute.defaultAttr().withName("age").withSparsity(0.8), NominalAttribute.defaultAttr().withName("size").withValues("small", "medium", "large"), BinaryAttribute.defaultAttr().withName("clicked").withValues("no", "yes"), NumericAttribute.defaultAttr(), NumericAttribute.defaultAttr() }; AttributeGroup group = new AttributeGroup("user", attrs); Assert.assertEquals(8, group.size()); Assert.assertEquals("user", group.name()); Assert.assertEquals(NumericAttribute.defaultAttr().withIndex(0), group.getAttr(0)); Assert.assertEquals(3, group.indexOf("age")); Assert.assertFalse(group.hasAttr("abc")); Assert.assertEquals(group, AttributeGroup.fromStructField(group.toStructField())); } }
@Test public void testAttributeGroup() { Attribute[] attrs = new Attribute[]{ NumericAttribute.defaultAttr(), NominalAttribute.defaultAttr(), BinaryAttribute.defaultAttr().withIndex(0), NumericAttribute.defaultAttr().withName("age").withSparsity(0.8), NominalAttribute.defaultAttr().withName("size").withValues("small", "medium", "large"), BinaryAttribute.defaultAttr().withName("clicked").withValues("no", "yes"), NumericAttribute.defaultAttr(), NumericAttribute.defaultAttr() }; AttributeGroup group = new AttributeGroup("user", attrs); Assert.assertEquals(8, group.size()); Assert.assertEquals("user", group.name()); Assert.assertEquals(NumericAttribute.defaultAttr().withIndex(0), group.getAttr(0)); Assert.assertEquals(3, group.indexOf("age")); Assert.assertFalse(group.hasAttr("abc")); Assert.assertEquals(group, AttributeGroup.fromStructField(group.toStructField())); } }
@Test public void testAttributeGroup() { Attribute[] attrs = new Attribute[]{ NumericAttribute.defaultAttr(), NominalAttribute.defaultAttr(), BinaryAttribute.defaultAttr().withIndex(0), NumericAttribute.defaultAttr().withName("age").withSparsity(0.8), NominalAttribute.defaultAttr().withName("size").withValues("small", "medium", "large"), BinaryAttribute.defaultAttr().withName("clicked").withValues("no", "yes"), NumericAttribute.defaultAttr(), NumericAttribute.defaultAttr() }; AttributeGroup group = new AttributeGroup("user", attrs); Assert.assertEquals(8, group.size()); Assert.assertEquals("user", group.name()); Assert.assertEquals(NumericAttribute.defaultAttr().withIndex(0), group.getAttr(0)); Assert.assertEquals(3, group.indexOf("age")); Assert.assertFalse(group.hasAttr("abc")); Assert.assertEquals(group, AttributeGroup.fromStructField(group.toStructField())); } }