federer/dist/src/coordinator/evaluation/TensorBoardOutput.d.ts

Experiments in asynchronous federated learning and decentralized learning

import * as tf from "@tensorflow/tfjs-node";
import { SummaryFileWriter } from "@tensorflow/tfjs-node/dist/tensorboard";
import { Logger } from "winston";
import { RoundSummary } from "../../common";
import { RoundResults } from "..";
import { CoordinatorOptions } from "../options/coordinator";
/**
 * Represents a TensorBoard output file to which we can write.
 */
export declare class TensorBoardOutput {
    protected readonly runName: string;
    protected readonly options: CoordinatorOptions;
    protected readonly logger: Logger;
    /** Writes to summary files that can be visualized in TensorBoard. */
    protected readonly summaryFileWriter: SummaryFileWriter;
    /** Path to the results directory */
    readonly resultsPath: string;
    constructor(runName: string, options: CoordinatorOptions, logger: Logger);
    /** Writes data distrbution to summary file*/
    writeDataVisualization(dataDistibution: tf.Tensor2D): void;
    /** Write the graphs showing evaluation results. */
    writeEvaluationResults(results: RoundResults): void;
    /**
     * Write the graphs that can be written from a round summary without
     * evaluating the results.
     *
     * @see {@link Coordinator.evaluate} for the graphs that result from evaluation
     */
    writeRoundSummary(summary: RoundSummary): void;
    private writeStalenessGraph;
}
//# sourceMappingURL=TensorBoardOutput.d.ts.map