Ops added: shape and one not encoding

Author: kailashahirwar

ravop/core/__init__.py

@@ -203,6 +203,9 @@ def slice(self, **kwargs):
     def find_indices(self, op, **kwargs):
         return find_indices(self, op, **kwargs)
 
+    def shape_(self, **kwargs):
+        return shape(self, **kwargs)
+
     # Comparison Ops
     def greater(self, op1, **kwargs):
         return greater(self, op1, **kwargs)
@@ -326,6 +329,8 @@ def __str__(self):
                                                                                           self._op_db.operator,
                                                                                           self.output,
                                                                                           self.status)
+    def __call__(self, *args, **kwargs):
+        return self.output
 
 
 class Scalar(Op):
@@ -368,6 +373,9 @@ def __str__(self):
         return "Scalar Op:\nId:{}\nOutput:{}\nStatus:{}\nDtype:{}\n".format(self.id, self.output,
                                                                             self.status, self.dtype)
 
+    def __call__(self, *args, **kwargs):
+        return self.output
+
 
 class Tensor(Op):
     """
@@ -411,6 +419,9 @@ def __str__(self):
         return "Tensor Op:\nId:{}\nOutput:{}\nStatus:{}\nDtype:{}\n".format(self.id, self.output,
                                                                             self.status, self.dtype)
 
+    def __call__(self, *args, **kwargs):
+        return self.output
+
 
 class Data(object):
     def __init__(self, id=None, value=None, dtype=None, **kwargs):
@@ -471,6 +482,9 @@ def dtype(self):
     def __str__(self):
         return "Data:\nId:{}\nDtype:{}\n".format(self.id, self.dtype)
 
+    def __call__(self, *args, **kwargs):
+        return self.value
+
 
 class Graph(object):
     """A class to represent a graph object"""
@@ -707,6 +721,10 @@ def find_indices(op, op2, **kwargs):
     return __create_math_op(op, op2, Operators.FIND_INDICES.value, **kwargs)
 
 
+def shape(op, **kwargs):
+    return __create_math_op2(op, Operators.SHAPE.value, **kwargs)
+
+
 # Comparison
 def greater(op1, op2, **kwargs):
     return __create_math_op(op1, op2, Operators.GREATER.value, **kwargs)
@@ -797,6 +815,12 @@ def sign(op1, **kwargs):
 def foreach(op, **kwargs):
     return __create_math_op2(op, Operators.FOREACH.value, **kwargs)
 
+# Data Preprocessing
+
+
+def one_hot_encoding(op, **kwargs):
+    return __create_math_op2(op, Operators.ONE_HOT_ENCODING.value, **kwargs)
+
 
 def __create_math_op(op1, op2, operator, **kwargs):
     if op1 is None or op2 is None:

ravop/enums.py

@@ -53,6 +53,7 @@ class Operators(Enum):
     TILE = "tile"
     SLICE = "slice"
     FIND_INDICES = "find_indices"
+    SHAPE = "shape"
 
     # Comparison Operators
     GREATER = "greater"
@@ -83,6 +84,9 @@ class Operators(Enum):
     SIGN = "sign"
     FOREACH = "foreach"
 
+    # Data Preprocessing
+    ONE_HOT_ENCODING = "one_hot_encoding"
+
 
 class TFJSOperators(Enum):
     SIGMOID = "sigmoid"