在ggplot中绘制神经网络图

library(ggplot2)

# 准备节点和连接信息
nodes <- data.frame(
  name = c("Input", "Hidden", "Output"),
  x = c(1, 2, 3),
  y = c(1, 2, 1)
)

connections <- data.frame(
  from = c("Input", "Hidden"),
  to = c("Hidden", "Output")
)

# 绘制神经网络图
ggplot() +
  geom_segment(data = connections, aes(x = nodes$x[match(from, nodes$name)],
                                       y = nodes$y[match(from, nodes$name)],
                                       xend = nodes$x[match(to, nodes$name)],
                                       yend = nodes$y[match(to, nodes$name)]),
               arrow = arrow(length = unit(0.3, "cm")),
               size = 1) +
  geom_text(data = nodes, aes(x = x, y = y, label = name),
            hjust = 0.5, vjust = 0.5, size = 5) +
  xlim(0, 4) +
  ylim(0, 3) +
  theme_void()