webshap/src/utils/math-import.ts

Explain any ML models anywhere

// We only import a small subset of Math.js functions

import {
  ones,
  zeros,
  diag,
  matrix,
  index,
  range,
  row,
  column,
  flatten,
  sum,
  mean,
  transpose,
  add,
  multiply,
  inv,
  pinv,
  det,
  reshape,
  squeeze
} from 'mathjs';

export default {
  ones,
  zeros,
  diag,
  matrix,
  index,
  range,
  row,
  column,
  flatten,
  sum,
  mean,
  transpose,
  add,
  multiply,
  inv,
  pinv,
  det,
  reshape,
  squeeze
};