Как ссылаться на объект NN из объекта Train?

Question

Как ссылаться на объект NN из объекта поезда? Мне нужно построить объект NN, чтобы увидеть диаграмму NN. Но приведенный ниже код не работает.

# Train and test the model with the number of folds
set.seed(123)

# Set the number of hidden layers
tune.grid.neuralnet <- expand.grid(
  layer1 = 2,
  layer2 = 2,
  layer3 = 2
)

train_obj <- caret::train(isFraud~amount+newbalanceDest+newbalanceOrig+oldbalanceDest+oldbalanceOrg+tran_ct,data=df_sample_train, method="neuralnet", act.fct = "logistic", linear.output=FALSE, tuneGrid = tune.grid.neuralnet, metric = "RMSE", trControl=trainCtrl)

summary(train_obj)
nn_obj = predict(train_obj,df_sample_test)
plot(nn_obj, rep="best")