Initial commit: ER-TP-DGP research prototype
Event-Reified Temporal Provenance Dual-Granularity Prompting for LLM-based APT detection on DARPA provenance datasets. Includes phase 0-14 method spec, IR/graph/metapath/trimming/prompt modules, scripts for THEIA candidate universe, landmark CSG construction, hybrid prompting, and LLM inference. Excludes data/, reports/, and local LLM config from version control.
This commit is contained in:
35
pyproject.toml
Normal file
35
pyproject.toml
Normal file
@@ -0,0 +1,35 @@
|
||||
[project]
|
||||
name = "er-tp-dgp"
|
||||
version = "0.1.0"
|
||||
description = "Event-Reified Temporal Provenance Dual-Granularity Prompting for LLM-based APT detection"
|
||||
requires-python = ">=3.10"
|
||||
readme = "README.md"
|
||||
license = { text = "Research prototype" }
|
||||
authors = [{ name = "ER-TP-DGP collaborators" }]
|
||||
dependencies = ["PyYAML>=6.0"]
|
||||
|
||||
[project.optional-dependencies]
|
||||
dev = ["pytest>=7.0"]
|
||||
local = [
|
||||
"torch>=2.3",
|
||||
"transformers>=4.45",
|
||||
"peft>=0.12",
|
||||
"accelerate>=0.34",
|
||||
"bitsandbytes>=0.43",
|
||||
"datasets>=2.20",
|
||||
"numpy>=1.26",
|
||||
]
|
||||
embed = [
|
||||
"sentence-transformers>=3.0",
|
||||
"numpy>=1.26",
|
||||
]
|
||||
eval = [
|
||||
"scikit-learn>=1.4",
|
||||
]
|
||||
|
||||
[tool.pytest.ini_options]
|
||||
pythonpath = [".", "src"]
|
||||
testpaths = ["tests"]
|
||||
|
||||
[tool.ruff]
|
||||
line-length = 100
|
||||
Reference in New Issue
Block a user