Updating lib.rs in delta_nn

delta-rs · Nov 26, 2024 · 2c1334e · 2c1334e
1 parent e03b245
commit 2c1334e
Showing 1 changed file with 5 additions and 13 deletions.
diff --git a/delta_nn/src/lib.rs b/delta_nn/src/lib.rs
@@ -1,14 +1,6 @@
-pub fn add(left: u64, right: u64) -> u64 {
-    left + right
-}
+pub mod models;
+pub mod layers;
 
-#[cfg(test)]
-mod tests {
-    use super::*;
-
-    #[test]
-    fn it_works() {
-        let result = add(2, 2);
-        assert_eq!(result, 4);
-    }
-}
+pub use models::sequential::Sequential;
+pub use layers::dense::Dense;
+pub use layers::relu::Relu;