federer
Version:
Experiments in asynchronous federated learning and decentralized learning
21 lines • 870 B
TypeScript
import * as tf from "@tensorflow/tfjs-node";
import { InitializerIdentifier } from "@tensorflow/tfjs-layers/dist/initializers";
import { OptimizerOptions } from "../../../coordinator";
export declare type LeafModelOptions = LSTMModelOptions | NNModelOptions;
interface SharedModelOptions {
optimizer: OptimizerOptions;
numberOutputClasses: number;
}
export interface LSTMModelOptions extends SharedModelOptions {
dataset: "shakespeare";
recurrentInitializer?: InitializerIdentifier;
}
export interface NNModelOptions extends SharedModelOptions {
dataset: "synthetic";
}
export declare function createModel(options: LeafModelOptions): tf.Sequential;
export declare function getModel(options: LeafModelOptions): tf.Sequential;
export declare const SEQUENCE_LENGTH = 80;
export declare const FEATURES = 60;
export {};
//# sourceMappingURL=model.d.ts.map