Function csvTransform

baseball-node/strike_zone.js:41–48

({xs, ys})

// Converts a row from the CSV into features and labels.
// Each feature field is normalized within training data constants:
const csvTransform = ({xs, ys}) => {
  const values = [
    normalize(xs.px, PX_MIN, PX_MAX), normalize(xs.pz, PZ_MIN, PZ_MAX),
    normalize(xs.sz_top, SZ_TOP_MIN, SZ_TOP_MAX),
    normalize(xs.sz_bot, SZ_BOT_MIN, SZ_BOT_MAX), xs.left_handed_batter
  ];
  return {xs: values, ys: ys.is_strike};
};

const trainingData =
    tf.data.csv(TRAIN_DATA_PATH, {columnConfigs: {is_strike: {isLabel: true}}})