[package]
edition = "2021"
name = "node2vec-rs"
version = "0.1.6"
authors = ["Gregor Lueg <gregorlueg@me.com>"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "A node2vec implementation in Rust via the Burn tensor framework."
readme = "README.md"
license = "MIT"
repository = "https://github.com/GregorLueg/node2vec-rs"
[features]
burn = ["dep:burn"]
cpu = []
default = ["cpu"]
metal = [
"burn",
"burn/wgpu",
]
ndarray = [
"burn",
"burn/ndarray",
]
ndarray-blas-accelerate = [
"burn",
"burn/ndarray",
]
ndarray-blas-openblas = [
"burn",
"burn/ndarray",
]
tch-cpu = [
"burn",
"burn/tch",
]
tch-mps = [
"burn",
"burn/tch",
]
vulkan = [
"burn",
"burn/wgpu",
]
wgpu = [
"burn",
"burn/wgpu",
]
[lib]
name = "node2vec_rs"
path = "src/lib.rs"
[[bin]]
name = "node2vec"
path = "src/main.rs"
[[test]]
name = "integration_tests"
path = "tests/integration_tests.rs"
[dependencies.burn]
version = "0.20.1"
features = [
"dataset",
"train",
]
optional = true
[dependencies.clap]
version = "4.5.53"
features = ["derive"]
[dependencies.csv]
version = "1.4.0"
[dependencies.faer]
version = "0.23.2"
[dependencies.indicatif]
version = "0.18.3"
[dependencies.rand]
version = "0.9.2"
[dependencies.rand_distr]
version = "0.5.1"
[dependencies.rayon]
version = "1.11.0"
[dependencies.rustc-hash]
version = "2.1.1"
[dependencies.tempfile]
version = "3.23.0"
[dependencies.thousands]
version = "0.2.0"
[dependencies.wide]
version = "1.1.1"
[profile.release]
opt-level = 3
lto = true
codegen-units = 1
strip = true