ch/Caps-Train
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Extract session parameters

Browse Source
This commit is contained in:
Christoph Hagen 2023-10-24 10:38:24 +02:00
parent 2913878b3e
commit a328c3a59e
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
Sources/ClassifierCreator.swift
View File

@ -4,6 +4,11 @@ import Combine
final class ClassifierCreator {
private let scenePrintRevision: Int? = 2
private let sessionProgressIterations = 10
private let sessionCheckpointIterations = 100
private let sessionIterations = 1000
let server: URL
let configuration: Configuration
@ -276,13 +281,15 @@ final class ClassifierCreator {
func trainModelAsync() async throws -> MLImageClassifier {
let params = MLImageClassifier.ModelParameters(
maxIterations: configuration.trainingIterations,
augmentation: [])
augmentation: [],
algorithm: .transferLearning(
featureExtractor: .scenePrint(revision: scenePrintRevision),
classifier: .logisticRegressor))
let sessionParameters = MLTrainingSessionParameters(
sessionDirectory: sessionDirectory,
reportInterval: 10,
checkpointInterval: 100,
iterations: 1000)
reportInterval: sessionProgressIterations,
checkpointInterval: sessionCheckpointIterations,
iterations: sessionIterations)
var subscriptions = [AnyCancellable]()