a8620b9e034f4e45138c253dd78709775f8e5c63,examples/simple_but_ugly/tf_models.py,MyModel,_build,#MyModel#,16

Before Change

class MyModel(TFModel):
    An example of a tf model class 
    def _build(self, *args, **kwargs):
        num_features = 3
        num_classes = 3
        x = tf.placeholder("float", [None, num_features], name="x")
        y = tf.placeholder("int32",[None], name="y")
        y_oe = tf.one_hot(y, num_classes)

        w = tf.Variable(tf.zeros([num_features,num_classes]))
        b = tf.Variable(tf.zeros([num_classes]))

        y_ = tf.nn.softmax(tf.matmul(x, w) + b)

        // Define a cost function
        tf.losses.add_loss(tf.losses.softmax_cross_entropy(y_oe, y_))

        print("___________________ MyModel initialized")

After Change

class MyModel(TFModel):
    An example of a tf model class 
    def _build(self, *args, **kwargs):
        images_shape = kwargs.get("images_shape")
        num_features =  images_shape[-1] if images_shape is not None else 3
        num_classes = kwargs.get("num_classes", 3)

        x = tf.placeholder("float", [None, num_features], name="x")
        y = tf.placeholder("int32",[None], name="y")

Instances