Method
__init__
(self, depth=None, on_value=None, off_value=None, axis=None, dtype=None, name=None)
Source from the content-addressed store, hash-verified
| 53 | """ |
| 54 | |
| 55 | def __init__(self, depth=None, on_value=None, off_value=None, axis=None, dtype=None, name=None): #'input'): |
| 56 | |
| 57 | super(OneHot, self).__init__(name) |
| 58 | self.depth = depth |
| 59 | self.on_value = on_value |
| 60 | self.off_value = off_value |
| 61 | self.axis = axis |
| 62 | self.dtype = dtype |
| 63 | logging.info("OneHotInput %s" % (self.name)) |
| 64 | |
| 65 | if not self._built: |
| 66 | self.build(tuple()) |
| 67 | self._built = True |
| 68 | |
| 69 | if self.depth is None: |
| 70 | raise RuntimeError(self.__class__.__name__ + ": depth == None the number of output units is undefined") |
| 71 | |
| 72 | def __repr__(self): |
| 73 | s = ('{classname}(depth={depth}') |
Tested by
no test coverage detected