predict.js

'use strict';

const path = require('path');
const { cwd, pipeline, output } = require('./metadata.json');

function _requirePlugin(name) {
  const mod = require(path.join(cwd, pipeline[name]));
  if (mod && typeof mod.default === 'function') {
    return mod.default;
  }
  return mod;
}

const modelDefineModule = _requirePlugin('modelDefine');

let dataProcessModule;
if (typeof pipeline.dataProcess === 'string') {
  dataProcessModule = _requirePlugin('dataProcess');
}

const model = modelDefineModule(null, {
  recoverPath: __dirname + '/model'
});

function predict(data) {
  const sample = { data, label: null };
  let future = model;

  if (typeof dataProcessModule === 'function') {
    future = future.then((m) => {
      dataProcessModule(sample, {}, pipeline.dataProcessParams);
      return m
    });
  }
  return future.then((m) => {
    m.predict(sample);
  });
};

module.exports = predict;