export function columnsAndFieldsFromYPred()

in modules/manifold/src/utils/data-processor.js [232:255]

• 21 lines of code
• 2 McCabe index (conditional complexity)

export function columnsAndFieldsFromYPred(yPred, classLabels) {
  const columns = [];
  const fields = [];

  // `classLabels` passed as argument can ensure correct order
  const _classLabels = classLabels || Object.keys(yPred[0][0]);

  yPred.forEach((singleModelPredArr, modelId) => {
    _classLabels.forEach((classCol, classId) => {
      const name = predColName(modelId, classId);
      columns.push(singleModelPredArr.map(predObj => predObj[classCol]));
      fields.push({
        name,
        type: FEATURE_TYPE.NUMERICAL,
        role: FIELD_ROLE.PREDICTION,
        dataType: DATA_TYPE.REAL,
      });
    });
  });
  return {
    columns,
    fields,
  };
}