arc-ml v0.0.3

Getting Started

This package is used for low-level control of the machine learning models that is used in arc-agents.

Installation

To start using the package, you can install it as follows:

npm install arc-ml

Model Initialization

Initializing a neural network with 1 action head

const { NeuralNetworkMultihead } =  require("arc-ml")

const metadata = {
  inputDim: 5,
  neurons: [16, 8, 4],
  activationFunctionName: "elu",
  actionNames: ["Actions"],
  actionOrder: [["action 1", "action 2"]],
  actionPolicies: ["probabilisticSampling"]
}

const model = new NeuralNetworkMultihead({ metadata }, true)

Initializing a tabular model with 2 action heads

const { FrequencyTable } =  require("arc-ml")

const metadata = {
  inputDim: 10,
  actionNames: ["Direction", "Combat"],
  actionOrder: [["direction 1", "direction 2"], ["combat 1", "combat 2"]],
  actionPolicies: ["argmaxPolicy", "probabilisticSampling"]
}

const model = new FrequencyTable({ metadata }, true)