[package]
edition = "2021"
name = "node2vec-rs"
version = "0.1.2"
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]
default = ["tch-cpu"]
metal = ["burn/metal"]
ndarray = ["burn/ndarray"]
ndarray-blas-accelerate = [
"burn/ndarray",
"burn/accelerate",
]
ndarray-blas-openblas = [
"burn/ndarray",
"burn/openblas",
]
tch-cpu = ["burn/tch"]
tch-mps = ["burn/tch"]
vulkan = ["burn/vulkan"]
wgpu = ["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",
]
[dependencies.clap]
version = "4.5.53"
features = ["derive"]
[dependencies.csv]
version = "1.4.0"
[dependencies.indicatif]
version = "0.18.3"
[dependencies.rand]
version = "0.9.2"
[dependencies.rayon]
version = "1.11.0"
[dependencies.rustc-hash]
version = "2.1.1"
[dependencies.tempfile]
version = "3.23.0"