tfjs-react
Version:
Various React components for tensorflow.js
35 lines • 1.47 kB
JavaScript
;
var __extends = (this && this.__extends) || (function () {
var extendStatics = function (d, b) {
extendStatics = Object.setPrototypeOf ||
({ __proto__: [] } instanceof Array && function (d, b) { d.__proto__ = b; }) ||
function (d, b) { for (var p in b) if (b.hasOwnProperty(p)) d[p] = b[p]; };
return extendStatics(d, b);
};
return function (d, b) {
extendStatics(d, b);
function __() { this.constructor = d; }
d.prototype = b === null ? Object.create(b) : (__.prototype = b.prototype, new __());
};
})();
Object.defineProperty(exports, "__esModule", { value: true });
var tfvis = require("@tensorflow/tfjs-vis");
var React = require("react");
var ModelSummary = /** @class */ (function (_super) {
__extends(ModelSummary, _super);
function ModelSummary(props) {
var _this = _super.call(this, props) || this;
_this._summaryElement = React.createRef();
return _this;
}
ModelSummary.prototype.componentDidMount = function () {
return tfvis.show.modelSummary(this._summaryElement.current, this.props.model);
};
ModelSummary.prototype.render = function () {
return (React.createElement(React.Fragment, null,
React.createElement("div", { ref: this._summaryElement })));
};
return ModelSummary;
}(React.Component));
exports.ModelSummary = ModelSummary;
//# sourceMappingURL=model-summary.js.map