Huge News!Announcing our $40M Series B led by Abstract Ventures.Learn More
Socket
Sign inDemoInstall
Socket

@tensorflow/tfjs-node

Package Overview
Dependencies
Maintainers
11
Versions
138
Alerts
File Explorer

Advanced tools

Socket logo

Install Socket

Detect and block malicious and high-risk dependencies

Install

@tensorflow/tfjs-node

**This repo is under active development and is not production-ready. We are actively developing as an open source project.**

  • 0.1.2
  • Source
  • npm
  • Socket score

Version published
Weekly downloads
38K
increased by6.06%
Maintainers
11
Weekly downloads
 
Created
Source

TensorFlow backend for TensorFlow.js via Node.js

This repo is under active development and is not production-ready. We are actively developing as an open source project.

Installing

npm install @tensorflow/tfjs-node
(or)
yarn add @tensorflow/tfjs-node

Before executing any TensorFlow.js code, load and set the backend to 'tensorflow'.

import * as tf from '@tensorflow/tfjs';
import '@tensorflow/tfjs-node';

tf.setBackend('tensorflow');

Development

# Download and install JS dependencies, including libtensorflow 1.8.
yarn

# Run TFJS tests against Node.js backend:
yarn test

See the demo directory that trains MNIST using TensorFlow.js with the TensorFlow C backend.

cd demo/
yarn

# Run the training script. See demo/package.json for this script.
yarn mnist

The important line to note is at the top of mnist.ts, which sets the backend to TensorFlow.

Optional: Build libtensorflow From TensorFlow source

This requires installing bazel first.

bazel build //tensorflow/tools/lib_package:libtensorflow

Supported Platforms

  • Mac OS
  • Linux
  • Windows coming soon

FAQs

Package last updated on 08 May 2018

Did you know?

Socket

Socket for GitHub automatically highlights issues in each pull request and monitors the health of all your open source dependencies. Discover the contents of your packages and block harmful activity before you install or update your dependencies.

Install

Related posts

SocketSocket SOC 2 Logo

Product

  • Package Alerts
  • Integrations
  • Docs
  • Pricing
  • FAQ
  • Roadmap
  • Changelog

Packages

npm

Stay in touch

Get open source security insights delivered straight into your inbox.


  • Terms
  • Privacy
  • Security

Made with ⚡️ by Socket Inc