| 260 | |
| 261 | #[test] |
| 262 | fn test_nn() { |
| 263 | |
| 264 | let n = NeuralNetwork::new(); |
| 265 | assert_eq!(n.layers.len(), 0); |
| 266 | assert_eq!(n.params.len(), 0); |
| 267 | |
| 268 | let b = NeuralNetwork::new().add_layer(3); |
| 269 | assert_eq!(b.layers, [3].to_vec()); |
| 270 | assert_eq!(b.params.len(), 0); |
| 271 | |
| 272 | let a = NeuralNetwork::new().add_layer(4).add_layer(3); |
| 273 | assert_eq!(a.layers, [4, 3].to_vec()); |
| 274 | assert_eq!(a.params.len(), 1); |
| 275 | assert_eq!(a.params[0].rows(), 3); |
| 276 | assert_eq!(a.params[0].cols(), 4); |
| 277 | |
| 278 | let c = NeuralNetwork::new().add_layer(4).add_layer(6).add_layer(11); |
| 279 | assert_eq!(c.layers, [4, 6, 11].to_vec()); |
| 280 | assert_eq!(c.params.len(), 2); |
| 281 | assert!(c.params[0].rows() == 6 && c.params[0].cols() == 4); |
| 282 | assert!(c.params[1].rows() == 11 && c.params[1].cols() == 7); |
| 283 | } |
| 284 | |
| 285 | #[test] |
| 286 | fn test_sigmoid() { |