aksquare.js

A deep learning library built using javascript to build and train machine learning models in the browsers.

$3

``
npm install aksquare.js --save
`

$3

via Script tag
Add the following code to an HTML file:

`











`
Via NPM

Install aksquare.js in your project using yarn or npm

`
import * as aksquare from 'aksquare.js'

let embed_dim = 50;
let model = new aksquare.Sequential([
new aksquare.Linear(n_words,embed_dim),
new aksquare.Linear(embed_dim,n_words),
new aksquare.Softmax()
]);

let optim = new aksquare.OptimSGD(model,lr=0.001);
let x_data = data[j]
let y_data = label[j]
let x = new aksquare.Tensor(1,n_words, false);
x.setFrom(x_data)

model.forward(x)

let loss = new aksquare.Loss(y_data-1,model);
total_loss += loss.out
loss.backward()
optim.step();
optim.grad_zero()
}
let embed_weight = get_weight(model.models[0].W)
console.log(embed_weight[0].length)
``

For contribution and bugs contact us : harishsg99@gmail.com