Skip to content

Commit

Permalink
Refactor serialization
Browse files Browse the repository at this point in the history
  • Loading branch information
DrA1ex committed Sep 19, 2023
1 parent 68da030 commit 49684b1
Show file tree
Hide file tree
Showing 13 changed files with 463 additions and 438 deletions.
2 changes: 1 addition & 1 deletion src/app/neural-network/engine/activations.ts
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import {IActivation, ISingleValueActivation} from "./base";
import {Matrix1D} from "./matrix";
import {Param} from "../serialization";
import {Param} from "../serialization/base";
import * as Matrix from "./matrix";
import * as Iter from "./iter";

Expand Down
2 changes: 1 addition & 1 deletion src/app/neural-network/engine/layers.ts
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ import * as matrix from "./matrix";
import {IActivation, ILayer, InitializerFn} from "./base";
import {ActivationsMap, ActivationT} from "./activations";
import {Initializers, InitializerT} from "./initializers";
import {Param} from "../serialization";
import {Param} from "../serialization/base";
import {Matrix1D} from "./matrix";
import * as Matrix from "./matrix";

Expand Down
2 changes: 1 addition & 1 deletion src/app/neural-network/engine/loss.ts
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
import {ILoss} from "./base";
import {Param} from "../serialization";
import {Param} from "../serialization/base";
import {Matrix1D} from "./matrix";
import * as Matrix from "./matrix";
import * as CommonUtils from "../utils/common"
Expand Down
2 changes: 1 addition & 1 deletion src/app/neural-network/engine/optimizers.ts
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import {Param} from "../serialization";
import {Param} from "../serialization/base";
import {ILayer, IOptimizer} from "./base";
import {Matrix1D, Matrix2D} from "./matrix";
import * as matrix from "./matrix";
Expand Down
4 changes: 2 additions & 2 deletions src/app/neural-network/neural-network.ts
Original file line number Diff line number Diff line change
Expand Up @@ -53,8 +53,8 @@ export {
} from "./chart"

export {
ModelSerialization, GanSerialization, ChainSerialization, UniversalModelSerializer
} from "./serialization"
ModelSerialization, ChainSerialization, UniversalModelSerializer, GanSerialization
} from "./serialization";

export {
ParallelModelWrapper, ParallelUtils
Expand Down

0 comments on commit 49684b1

Please sign in to comment.