R: как округлить цифры веса в plot.nn ()? - PullRequest
2 голосов
/ 30 мая 2020

Я пытаюсь построить свою нейронную сеть, и мне интересно, как мне округлить веса до трех цифр.

library(neuralnet)
set.seed(0)

x = matrix(rnorm(100, 0, 5), ncol=4)
y = rnorm(25, 100, 20)
data = data.frame(y, x)
nn.model = neuralnet(y~., data, linear.output=T, stepmax = 1e+06)

plot(nn.model)

Я пробовал mapply(round), но это не сработало в списках как модель neuralnet генерирует. Любое предложение приветствуется!

1 Ответ

2 голосов
/ 30 мая 2020

Вот так:

nn.model$weights[[1]] <- lapply(nn.model$weights[[1]], function(x) round(x, 3))
plot(nn.model)

enter image description here

...