def test_prune(self): pa = feature.FeatureArray() p0 = feature.FeaturePair(1, 2) p1 = feature.FeaturePair(2, 1) p2 = feature.FeaturePair(3, 4) pa.appendFeaturePair(p0) pa.appendFeaturePair(p1) pa.appendFeaturePair(p2) pb = feature.FeatureArray() p0 = feature.FeaturePair(1, 2) p1 = feature.FeaturePair(2, 2) p2 = feature.FeaturePair(3, 2) pb.appendFeaturePair(p0) pb.appendFeaturePair(p1) pb.appendFeaturePair(p2) pruner = feature.Pruner() pruner.recordFeatureArray("a", pa) pruner.recordFeatureArray("b", pb) pruner.prune(5) for key, farr in pruner: sys.stdout.write(str(key) + " ") feature.printFeatureArray(farr)
def test_feature_conversion(self): manager = feature.FeatureManager() str0 = "hello world" str1 = "hello yongjoo" str2 = "hello hello good" fa0 = manager.convertTextIntoFeatureArray(str0) fa1 = manager.convertTextIntoFeatureArray(str1) fa2 = manager.convertTextIntoFeatureArray(str2) print 'first string' feature.printFeatureArray(fa0) print 'second string' feature.printFeatureArray(fa1) print 'third string' feature.printFeatureArray(fa2)