learning_model 1.0.5 → 1.0.14
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +15 -1
- package/dist/index.bundle.js +1 -1
- package/dist/types/learning/base.d.ts +3 -5
- package/dist/types/learning/image.d.ts +5 -2
- package/dist/types/learning/mobilenet_image.d.ts +5 -2
- package/dist/types/learning/mobilenet_image.test.d.ts +1 -0
- package/dist/types/public/index.d.ts +1 -0
- package/dist/types/src/index.d.ts +3 -0
- package/dist/types/src/learning/base.d.ts +19 -0
- package/dist/types/src/learning/image.d.ts +40 -0
- package/dist/types/src/learning/mobilenet_image.d.ts +42 -0
- package/package.json +5 -2
- package/public/index.ts +4 -4
- package/src/learning/base.ts +15 -11
- package/src/learning/image.ts +18 -9
- package/src/learning/mobilenet_image.test.ts +63 -0
- package/src/learning/mobilenet_image.ts +18 -11
- package/tsconfig.json +5 -3
- package/types/learning/base.d.ts +3 -5
- package/types/learning/image.d.ts +5 -2
- package/types/learning/mobilenet_image.d.ts +5 -2
- package/types/learning/mobilenet_image.test.d.ts +1 -0
- package/types/src/learning/base.d.ts +3 -5
- package/types/src/learning/image.d.ts +5 -2
- package/types/src/learning/mobilenet_image.d.ts +5 -2
|
@@ -1,19 +1,17 @@
|
|
|
1
1
|
import * as tf from '@tensorflow/tfjs';
|
|
2
2
|
interface LearningInterface {
|
|
3
3
|
model: tf.LayersModel | null;
|
|
4
|
-
epochs: number;
|
|
5
|
-
batchSize: number;
|
|
6
4
|
labels: string[];
|
|
7
5
|
isRunning: boolean;
|
|
8
6
|
isReady: boolean;
|
|
9
|
-
limitSize: number;
|
|
10
7
|
onProgress(progress: number): void;
|
|
11
8
|
onLoss(loss: number): void;
|
|
9
|
+
onEvents(logs: any): void;
|
|
12
10
|
onTrainBegin(log: any): void;
|
|
13
11
|
onTrainEnd(log: any): void;
|
|
14
|
-
addData(label: string, data: any): void
|
|
12
|
+
addData(label: string, data: any): Promise<void>;
|
|
15
13
|
train(): Promise<tf.History>;
|
|
16
|
-
infer(data: any): Promise<
|
|
14
|
+
infer(data: any): Promise<any>;
|
|
17
15
|
saveModel(): void;
|
|
18
16
|
running(): boolean;
|
|
19
17
|
ready(): boolean;
|
|
@@ -4,6 +4,7 @@ declare class LearningImage implements LearningInterface {
|
|
|
4
4
|
model: tf.LayersModel | null;
|
|
5
5
|
epochs: number;
|
|
6
6
|
batchSize: number;
|
|
7
|
+
learningRate: number;
|
|
7
8
|
labels: string[];
|
|
8
9
|
isRunning: boolean;
|
|
9
10
|
isReady: boolean;
|
|
@@ -13,17 +14,19 @@ declare class LearningImage implements LearningInterface {
|
|
|
13
14
|
readonly MOBILE_NET_INPUT_HEIGHT = 224;
|
|
14
15
|
readonly MOBILE_NET_INPUT_CHANNEL = 3;
|
|
15
16
|
readonly IMAGE_NORMALIZATION_FACTOR = 255;
|
|
16
|
-
constructor({ epochs, batchSize, limitSize, }?: {
|
|
17
|
+
constructor({ epochs, batchSize, limitSize, learningRate, }?: {
|
|
17
18
|
modelURL?: string;
|
|
18
19
|
epochs?: number;
|
|
19
20
|
batchSize?: number;
|
|
20
21
|
limitSize?: number;
|
|
22
|
+
learningRate?: number;
|
|
21
23
|
});
|
|
22
24
|
onProgress: (progress: number) => void;
|
|
23
25
|
onLoss: (loss: number) => void;
|
|
26
|
+
onEvents: (logs: any) => void;
|
|
24
27
|
onTrainBegin: (log: any) => void;
|
|
25
28
|
onTrainEnd: (log: any) => void;
|
|
26
|
-
addData(label: string, data: any): void
|
|
29
|
+
addData(label: string, data: any): Promise<void>;
|
|
27
30
|
train(): Promise<tf.History>;
|
|
28
31
|
infer(data: any): Promise<Map<string, number>>;
|
|
29
32
|
saveModel(): void;
|
|
@@ -4,6 +4,7 @@ declare class LearningMobilenetImage implements LearningInterface {
|
|
|
4
4
|
model: tf.LayersModel | null;
|
|
5
5
|
epochs: number;
|
|
6
6
|
batchSize: number;
|
|
7
|
+
learningRate: number;
|
|
7
8
|
labels: string[];
|
|
8
9
|
modelURL: string;
|
|
9
10
|
isRunning: boolean;
|
|
@@ -15,17 +16,19 @@ declare class LearningMobilenetImage implements LearningInterface {
|
|
|
15
16
|
readonly MOBILE_NET_INPUT_CHANNEL = 3;
|
|
16
17
|
readonly IMAGE_NORMALIZATION_FACTOR = 255;
|
|
17
18
|
constructor({ modelURL, // 디폴트 mobilenet 이미지
|
|
18
|
-
epochs, batchSize, limitSize, }?: {
|
|
19
|
+
epochs, batchSize, limitSize, learningRate, }?: {
|
|
19
20
|
modelURL?: string;
|
|
20
21
|
epochs?: number;
|
|
21
22
|
batchSize?: number;
|
|
22
23
|
limitSize?: number;
|
|
24
|
+
learningRate?: number;
|
|
23
25
|
});
|
|
24
26
|
onProgress: (progress: number) => void;
|
|
25
27
|
onLoss: (loss: number) => void;
|
|
28
|
+
onEvents: (logs: any) => void;
|
|
26
29
|
onTrainBegin: (log: any) => void;
|
|
27
30
|
onTrainEnd: (log: any) => void;
|
|
28
|
-
addData(label: string, data: any): void
|
|
31
|
+
addData(label: string, data: any): Promise<void>;
|
|
29
32
|
train(): Promise<tf.History>;
|
|
30
33
|
infer(data: any): Promise<Map<string, number>>;
|
|
31
34
|
saveModel(): void;
|