Skip to content

Commit

Permalink
refactor: line length
Browse files Browse the repository at this point in the history
  • Loading branch information
1zun4 committed Feb 16, 2025
1 parent 03a801c commit 4668a24
Showing 1 changed file with 6 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -92,9 +92,13 @@ object CommandTensorflow : CommandFactory {
val (dataset, time3) = measureTimedValue { prepareData(linearTrainingData + trainingData) }
chat("Prepared dataset with ${dataset.features.size} samples in ${time3.inWholeMilliseconds}ms")

val (modelYaw, time4) = measureTimedValue { createAndTrainModel(dataset.features, dataset.labelX) }
val (modelYaw, time4) = measureTimedValue {
createAndTrainModel(dataset.features, dataset.labelX)
}
chat("Trained yaw model in ${time4.inWholeSeconds}s")
val (modelPitch, time5) = measureTimedValue { createAndTrainModel(dataset.features, dataset.labelY) }
val (modelPitch, time5) = measureTimedValue {
createAndTrainModel(dataset.features, dataset.labelY)
}
chat("Trained pitch model in ${time5.inWholeSeconds}s")

// Save the models
Expand Down

0 comments on commit 4668a24

Please sign in to comment.