CausalNetModels
You can directly use an instance of this class.
causalNetModels
Extends:
* → CausalNetModels
This class provide common used models which can be accessed via causalNetModels instance.
Example:
import { SingleLabelClassification } from 'causal-net.models';
import { causalNetCore } from 'causal-net.core';
let model = new SingleLabelClassification(2);
let T = causalNetCore.CoreTensor;
let inputs = T.tensor([[0.1, 0.2]], [1, 2], 'float32');
let labels = T.tensor([[0, 1]], [1, 2], 'float32');
model.LayerRunner = { Predictor: (input)=>input};
model.Fit(inputs).print();
model.Loss(inputs, labels).print();
model.Predict(inputs).print();
model.OneHotPredict(inputs).print();Constructor Summary
| Public Constructor | ||
| public |
|
|
Member Summary
| Public Members | ||
| public |
model: * |
|
Method Summary
| Public Methods | ||
| public |
classification(numClass: *): * |
|
Public Constructors
public constructor() source
Public Members
public model: * source
Public Methods
public classification(numClass: *): * source
Params:
| Name | Type | Attribute | Description |
| numClass | * |
Return:
| * |