File size: 465 Bytes
0f27fb6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
"""Build the cleaned dataset with both split schemes.

    python scripts/01_build_dataset.py
"""

import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "src"))

from cti_attack import data  # noqa: E402


def main() -> None:
    records, labels, stats = data.build(verbose=True)
    print()
    data.write(records, labels, stats)
    print(f"\n{len(labels)} techniques retained")


if __name__ == "__main__":
    main()