Using tensorflow in armory
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
Sources/arm
Subprojects
.gitmodules
README.md
tensorflow.blend

README.md

tensorflow_example

tensorflow.js for Armory (Krom, HTML5).

Usage:

  • Recursive clone - git clone --recursive https://github.com/armory3d/tensorflow_example
  • Open .blend and run(F5)
  • Output - MyTrait.hx:19: Tensor [[8.2146339],]
package arm;

import tf.TF;
import tf.TFHelper;

class MyTrait extends iron.Trait {
	public function new() {
		super();

		notifyOnInit(function() {
			TFHelper.init(function() {
				var model = TF.sequential();
				model.add(TF.layers.dense({units: 1, inputShape: [1]}));
				model.compile({loss: 'meanSquaredError', optimizer: 'sgd'});
				var xs = TF.tensor2d([1, 2, 3, 4], [4, 1]);
				var ys = TF.tensor2d([1, 3, 5, 7], [4, 1]);
				model.fit(xs, ys).then(function() {
					var res = model.predict(TF.tensor2d([5], [1, 1]));
					trace(res);
				});
			});
		});
	}
}