[models] extract iris/

xieyuheng · May 30, 2024 · acc6fef · acc6fef
1 parent dccb57c
commit acc6fef
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 7 deletions.
diff --git a/TODO.md b/TODO.md
@@ -1,7 +1,7 @@
 # system-a
 
+[models] move `l2Loss` to `loss/`
 [models] test `argmax`
-[models] extract `iris/`
 [models] `compareClassification1`
 [models] `compareClassification` -- `extend2` on 1 1
 [models] `accuracy`

diff --git a/src/system-a/models/irisDataset.ts → src/system-a/models/iris/irisDataset.ts b/src/system-a/models/irisDataset.ts → src/system-a/models/iris/irisDataset.ts
diff --git a/src/system-a/models/irisModel.test.ts → src/system-a/models/iris/irisModel.test.ts b/src/system-a/models/irisModel.test.ts → src/system-a/models/iris/irisModel.test.ts
diff --git a/src/system-a/models/irisModel.ts → src/system-a/models/iris/irisModel.ts b/src/system-a/models/irisModel.ts → src/system-a/models/iris/irisModel.ts
@@ -1,10 +1,14 @@
-import { blockStack, denseBlock, denseInitParameters } from "../block/index.js"
-import { gradientDescentNaked } from "../gradient-descent/gradientDescentNaked.js"
-import { l2Loss } from "../loss.js"
-import type { Tensor } from "../tensor/Tensor.js"
-import { samplingObjective } from "../tensor/samplingObjective.js"
+import {
+  blockStack,
+  denseBlock,
+  denseInitParameters,
+} from "../../block/index.js"
+import { gradientDescentNaked } from "../../gradient-descent/gradientDescentNaked.js"
+import { l2Loss } from "../../loss.js"
+import type { Tensor } from "../../tensor/Tensor.js"
+import { samplingObjective } from "../../tensor/samplingObjective.js"
+import { model } from "../model.js"
 import { irisTrainXs, irisTrainYs } from "./irisDataset.js"
-import { model } from "./model.js"
 
 export const irisNetwork = blockStack([denseBlock(4, 6), denseBlock(6, 3)])