[package]
edition = "2021"
rust-version = "1.75"
name = "tranz"
version = "0.1.0"
authors = ["Arc <attobop@gmail.com>"]
build = false
publish = true
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Point-embedding knowledge graph models: TransE, RotatE, ComplEx. GPU training via candle."
readme = "README.md"
keywords = [
"knowledge-graph",
"embeddings",
"link-prediction",
]
categories = [
"science",
"algorithms",
]
license = "MIT OR Apache-2.0"
repository = "https://github.com/arclabs561/tranz"
[features]
candle = [
"dep:candle-core",
"dep:candle-nn",
]
cuda = [
"candle",
"candle-core/cuda",
]
default = ["rand"]
[lib]
name = "tranz"
path = "src/lib.rs"
[dependencies.candle-core]
version = "0.9"
optional = true
[dependencies.candle-nn]
version = "0.9"
optional = true
[dependencies.rand]
version = "0.9"
optional = true
[dependencies.serde]
version = "1"
features = ["derive"]
[dependencies.thiserror]
version = "2"
[dev-dependencies.proptest]
version = "1.5"
[dev-dependencies.tempfile]
version = "3"
[lints.rust]
missing_docs = "warn"