Add feature generation

neuralogic/__version__.py

@@ -1 +1 @@
-__version__ = "0.7.19"
+__version__ = "0.7.20"

neuralogic/core/constructs/java_objects.py

@@ -160,6 +160,16 @@ def atom_to_clause(self, atom):
 
         return self.clause(literal_array)
 
+    def to_clause(self, atoms):
+        literal_array = self.literal[len(atoms)]
+
+        for i, atom in enumerate(atoms):
+            terms = [self.get_term(term, self.variable_factory) for term in atom.terms]
+            predicate_name = f"_{atom.predicate.name}" if atom.predicate.hidden else atom.predicate.name
+
+            literal_array[i] = self.literal(predicate_name, atom.negated, terms)
+        return self.clause(literal_array)
+
     def get_generic_relation(self, relation_class, relation, variable_factory, default_weight=None, is_example=False):
         predicate = self.get_predicate(relation.predicate)
 

neuralogic/dataset/logic.py

@@ -1,5 +1,8 @@
 from typing import Optional, List, Union, Sequence
 
+import jpype
+
+from neuralogic.core.constructs.java_objects import JavaFactory
 from neuralogic.core.constructs.relation import BaseRelation
 from neuralogic.core.constructs.rule import Rule
 from neuralogic.dataset.base import BaseDataset
@@ -99,3 +102,17 @@ def set_examples(self, examples: List):
 
     def set_queries(self, queries: List):
         self._queries = queries
+
+    def generate_features(self, feature_depth: int = 1, count_groundings: bool = True):
+        java_factory = JavaFactory()
+        clause = jpype.java.util.ArrayList([java_factory.to_clause(sample.example) for sample in self.samples])
+
+        namespace = "cz.cvut.fel.ida.logic.features.generation"
+
+        jpype.JClass(f"{namespace}.FeatureGenerationSettings").COUNT_GROUNDINGS = count_groundings
+        features = jpype.JClass(f"{namespace}.FeatureGenerator").generateFeatures(clause, feature_depth)
+
+        table = [[int(i) for i in feats] for feats in features.table]
+        clauses = [str(clause) for clause in features.features]
+
+        return table, clauses