Size parameter in the nnet method

[sjevelazco]

Write a code to calculate a default value for the parameter size in the net method
It must be implemented in fit_nnet

nnet::nnet(
formula1,
data = train[[i]],
size = 2, # revise and implement a formula to calculate it
rang = 0.1,
# decay = grid$decay[ii],
maxit = 200,
trace = FALSE
)

https://stats.stackexchange.com/questions/181/how-to-choose-the-number-of-hidden-layers-and-nodes-in-a-feedforward-neural-netw#:~:text=However%2C%20neural%20networks%20with%20two,more%20than%20two%20hidden%20layers.