test

gistfile1.py

    def build(self):
        """
        Builds the full Keras model and stores it in self.model.
        """
        mc = self.config.model
        in_x = x = Input((1, 12, 54))

        x = Flatten(name='initial_flatten')(x)
        x = Dense(1024,activation='relu')(x)

        res_out = x

        # for policy out
        x = Dense(512, activation='relu')(res_out)
        policy_out = Dense(self.config.n_labels, activation="softmax",name="policy_out")(x)

        # for value output
        x = Dense(128, activation='relu')(res_out)
        value_out = Dense(1,name="value_out")(x)

        self.model = Model(in_x, [policy_out, value_out], name="chess_model")