Function gameToFeaturesAndLabel

data-generator/index.js:56–66

* Takes the state of one complete game and returns features suitable for * training. Returns an object containing features = player1's hand represented * using oneHot encoding, and label = whether player 1 won. * @param {*} gameState

(gameState)

 * @param {*} gameState
 */
function gameToFeaturesAndLabel(gameState) {
  return tf.tidy(() => {
    const player1Hand = tf.tensor1d(gameState.player1Hand, 'int32');
    const handOneHot = tf.oneHot(
        tf.sub(player1Hand, tf.scalar(1, 'int32')),
        game.GAME_STATE.max_card_value);
    const features = tf.sum(handOneHot, 0);
    const label = tf.tensor1d([gameState.player1Win]);
    return {xs: features, ys: label};
  });
}

/**
 * Collects one random play of the game.  Processes the sample to generate