Spaces:
Sleeping
Sleeping
“Transcendental-Programmer”
commited on
Commit
·
4932e02
1
Parent(s):
632f510
FEAT: Init project structure and basic config
Browse files- .gitignore +32 -119
- README.md +35 -2
- config/client_config.yaml +23 -0
- config/logging_config.yaml +19 -0
- config/server_config.yaml +18 -0
- dir_structure.md +60 -0
- docker/Dockerfile.client +24 -0
- docker/Dockerfile.server +27 -0
- docker/docker-compose.yml +40 -0
- docs/index.md +30 -0
- requirements.txt +36 -0
- src/__init__.py +2 -0
- src/client/__init__.py +2 -0
- src/models/__init__.py +2 -0
- src/rag/__init__.py +2 -0
- src/server/__init__.py +2 -0
- src/utils/__init__.py +2 -0
- tests/__init__.py +2 -0
.gitignore
CHANGED
@@ -1,12 +1,8 @@
|
|
1 |
-
#
|
2 |
__pycache__/
|
3 |
*.py[cod]
|
4 |
*$py.class
|
5 |
-
|
6 |
-
# C extensions
|
7 |
*.so
|
8 |
-
|
9 |
-
# Distribution / packaging
|
10 |
.Python
|
11 |
build/
|
12 |
develop-eggs/
|
@@ -20,106 +16,51 @@ parts/
|
|
20 |
sdist/
|
21 |
var/
|
22 |
wheels/
|
23 |
-
share/python-wheels/
|
24 |
*.egg-info/
|
25 |
.installed.cfg
|
26 |
*.egg
|
27 |
-
MANIFEST
|
28 |
|
29 |
-
#
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
|
|
|
|
|
|
|
|
34 |
|
35 |
-
#
|
36 |
-
|
37 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
38 |
|
39 |
# Unit test / coverage reports
|
40 |
htmlcov/
|
41 |
.tox/
|
42 |
-
.nox/
|
43 |
.coverage
|
44 |
.coverage.*
|
45 |
.cache
|
46 |
nosetests.xml
|
47 |
coverage.xml
|
48 |
*.cover
|
49 |
-
*.py,cover
|
50 |
.hypothesis/
|
51 |
-
.pytest_cache/
|
52 |
-
cover/
|
53 |
-
|
54 |
-
# Translations
|
55 |
-
*.mo
|
56 |
-
*.pot
|
57 |
-
|
58 |
-
# Django stuff:
|
59 |
-
*.log
|
60 |
-
local_settings.py
|
61 |
-
db.sqlite3
|
62 |
-
db.sqlite3-journal
|
63 |
-
|
64 |
-
# Flask stuff:
|
65 |
-
instance/
|
66 |
-
.webassets-cache
|
67 |
-
|
68 |
-
# Scrapy stuff:
|
69 |
-
.scrapy
|
70 |
-
|
71 |
-
# Sphinx documentation
|
72 |
-
docs/_build/
|
73 |
-
|
74 |
-
# PyBuilder
|
75 |
-
.pybuilder/
|
76 |
-
target/
|
77 |
-
|
78 |
-
# Jupyter Notebook
|
79 |
-
.ipynb_checkpoints
|
80 |
-
|
81 |
-
# IPython
|
82 |
-
profile_default/
|
83 |
-
ipython_config.py
|
84 |
-
|
85 |
-
# pyenv
|
86 |
-
# For a library or package, you might want to ignore these files since the code is
|
87 |
-
# intended to run in multiple environments; otherwise, check them in:
|
88 |
-
# .python-version
|
89 |
-
|
90 |
-
# pipenv
|
91 |
-
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
|
92 |
-
# However, in case of collaboration, if having platform-specific dependencies or dependencies
|
93 |
-
# having no cross-platform support, pipenv may install dependencies that don't work, or not
|
94 |
-
# install all needed dependencies.
|
95 |
-
#Pipfile.lock
|
96 |
-
|
97 |
-
# poetry
|
98 |
-
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
|
99 |
-
# This is especially recommended for binary packages to ensure reproducibility, and is more
|
100 |
-
# commonly ignored for libraries.
|
101 |
-
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
|
102 |
-
#poetry.lock
|
103 |
-
|
104 |
-
# pdm
|
105 |
-
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
|
106 |
-
#pdm.lock
|
107 |
-
# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
|
108 |
-
# in version control.
|
109 |
-
# https://pdm.fming.dev/latest/usage/project/#working-with-version-control
|
110 |
-
.pdm.toml
|
111 |
-
.pdm-python
|
112 |
-
.pdm-build/
|
113 |
-
|
114 |
-
# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
|
115 |
-
__pypackages__/
|
116 |
-
|
117 |
-
# Celery stuff
|
118 |
-
celerybeat-schedule
|
119 |
-
celerybeat.pid
|
120 |
-
|
121 |
-
# SageMath parsed files
|
122 |
-
*.sage.py
|
123 |
|
124 |
# Environments
|
125 |
.env
|
@@ -130,33 +71,5 @@ ENV/
|
|
130 |
env.bak/
|
131 |
venv.bak/
|
132 |
|
133 |
-
|
134 |
-
.
|
135 |
-
.spyproject
|
136 |
-
|
137 |
-
# Rope project settings
|
138 |
-
.ropeproject
|
139 |
-
|
140 |
-
# mkdocs documentation
|
141 |
-
/site
|
142 |
-
|
143 |
-
# mypy
|
144 |
-
.mypy_cache/
|
145 |
-
.dmypy.json
|
146 |
-
dmypy.json
|
147 |
-
|
148 |
-
# Pyre type checker
|
149 |
-
.pyre/
|
150 |
-
|
151 |
-
# pytype static type analyzer
|
152 |
-
.pytype/
|
153 |
-
|
154 |
-
# Cython debug symbols
|
155 |
-
cython_debug/
|
156 |
-
|
157 |
-
# PyCharm
|
158 |
-
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can
|
159 |
-
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
|
160 |
-
# and can be added to the global gitignore or merged into this file. For a more nuclear
|
161 |
-
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
162 |
-
#.idea/
|
|
|
1 |
+
# Python
|
2 |
__pycache__/
|
3 |
*.py[cod]
|
4 |
*$py.class
|
|
|
|
|
5 |
*.so
|
|
|
|
|
6 |
.Python
|
7 |
build/
|
8 |
develop-eggs/
|
|
|
16 |
sdist/
|
17 |
var/
|
18 |
wheels/
|
|
|
19 |
*.egg-info/
|
20 |
.installed.cfg
|
21 |
*.egg
|
|
|
22 |
|
23 |
+
# Virtual Environment
|
24 |
+
venv/
|
25 |
+
ENV/
|
26 |
+
|
27 |
+
# IDEs
|
28 |
+
.idea/
|
29 |
+
.vscode/
|
30 |
+
*.swp
|
31 |
+
*.swo
|
32 |
|
33 |
+
# Jupyter Notebook
|
34 |
+
.ipynb_checkpoints
|
35 |
+
|
36 |
+
# Distribution / packaging
|
37 |
+
.Python
|
38 |
+
build/
|
39 |
+
develop-eggs/
|
40 |
+
dist/
|
41 |
+
downloads/
|
42 |
+
eggs/
|
43 |
+
.eggs/
|
44 |
+
lib/
|
45 |
+
lib64/
|
46 |
+
parts/
|
47 |
+
sdist/
|
48 |
+
var/
|
49 |
+
wheels/
|
50 |
+
*.egg-info/
|
51 |
+
.installed.cfg
|
52 |
+
*.egg
|
53 |
|
54 |
# Unit test / coverage reports
|
55 |
htmlcov/
|
56 |
.tox/
|
|
|
57 |
.coverage
|
58 |
.coverage.*
|
59 |
.cache
|
60 |
nosetests.xml
|
61 |
coverage.xml
|
62 |
*.cover
|
|
|
63 |
.hypothesis/
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
64 |
|
65 |
# Environments
|
66 |
.env
|
|
|
71 |
env.bak/
|
72 |
venv.bak/
|
73 |
|
74 |
+
ideas.md
|
75 |
+
plan.md
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
README.md
CHANGED
@@ -1,2 +1,35 @@
|
|
1 |
-
#
|
2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Federated Learning for Privacy-Preserving Financial Data Generation with RAG Integration
|
2 |
+
|
3 |
+
This project implements a federated learning framework combined with a Retrieval-Augmented Generation (RAG) system to generate privacy-preserving synthetic financial data.
|
4 |
+
|
5 |
+
## Features
|
6 |
+
|
7 |
+
- Federated Learning using TensorFlow Federated
|
8 |
+
- Privacy-preserving data generation using VAE/GAN
|
9 |
+
- RAG integration for enhanced data quality
|
10 |
+
- Secure Multi-Party Computation (SMPC)
|
11 |
+
- Differential Privacy implementation
|
12 |
+
- Kubernetes-based deployment
|
13 |
+
- Comprehensive monitoring and logging
|
14 |
+
|
15 |
+
## Installation
|
16 |
+
|
17 |
+
```bash
|
18 |
+
pip install -r requirements.txt
|
19 |
+
```
|
20 |
+
|
21 |
+
## Usage
|
22 |
+
|
23 |
+
[Add usage instructions here]
|
24 |
+
|
25 |
+
## Project Structure
|
26 |
+
|
27 |
+
[Add project structure description here]
|
28 |
+
|
29 |
+
## License
|
30 |
+
|
31 |
+
MIT
|
32 |
+
|
33 |
+
## Contributing
|
34 |
+
|
35 |
+
[Add contributing guidelines here]
|
config/client_config.yaml
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# client_config.yaml configuration
|
2 |
+
|
3 |
+
client:
|
4 |
+
# Data configuration
|
5 |
+
data:
|
6 |
+
batch_size: 32
|
7 |
+
shuffle_buffer: 1000
|
8 |
+
input_dim: 32
|
9 |
+
|
10 |
+
# Model configuration
|
11 |
+
model:
|
12 |
+
type: "feedforward"
|
13 |
+
hidden_dims: [128, 64]
|
14 |
+
activation: "relu"
|
15 |
+
|
16 |
+
# Training configuration
|
17 |
+
training:
|
18 |
+
local_epochs: 5
|
19 |
+
learning_rate: 0.001
|
20 |
+
|
21 |
+
monitoring:
|
22 |
+
log_level: "INFO"
|
23 |
+
|
config/logging_config.yaml
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
version: 1
|
2 |
+
formatters:
|
3 |
+
detailed:
|
4 |
+
format: '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
5 |
+
handlers:
|
6 |
+
console:
|
7 |
+
class: logging.StreamHandler
|
8 |
+
level: INFO
|
9 |
+
formatter: detailed
|
10 |
+
stream: ext://sys.stdout
|
11 |
+
file:
|
12 |
+
class: logging.FileHandler
|
13 |
+
level: INFO
|
14 |
+
formatter: detailed
|
15 |
+
filename: logs/federated_learning.log
|
16 |
+
loggers:
|
17 |
+
'': # Root logger
|
18 |
+
level: INFO
|
19 |
+
handlers: [console, file]
|
config/server_config.yaml
ADDED
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# server_config.yaml configuration
|
2 |
+
|
3 |
+
server:
|
4 |
+
# Federated learning configuration
|
5 |
+
federated:
|
6 |
+
min_clients: 1
|
7 |
+
rounds: 10
|
8 |
+
sample_fraction: 0.8
|
9 |
+
|
10 |
+
# Aggregation configuration
|
11 |
+
aggregation:
|
12 |
+
method: "fedavg"
|
13 |
+
weighted: true
|
14 |
+
|
15 |
+
# Monitoring configuration
|
16 |
+
monitoring:
|
17 |
+
log_level: "INFO"
|
18 |
+
|
dir_structure.md
ADDED
@@ -0,0 +1,60 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Directory Structure
|
2 |
+
|
3 |
+
```
|
4 |
+
federated-rag-financial/
|
5 |
+
├── .github/
|
6 |
+
│ └── workflows/
|
7 |
+
│ └── ci.yml
|
8 |
+
├── .gitignore
|
9 |
+
├── docker/
|
10 |
+
│ ├── Dockerfile.client
|
11 |
+
│ ├── Dockerfile.server
|
12 |
+
│ └── docker-compose.yml
|
13 |
+
├── kubernetes/
|
14 |
+
│ ├── deployments/
|
15 |
+
│ │ ├── client.yaml
|
16 |
+
│ │ ├── server.yaml
|
17 |
+
│ │ └── rag.yaml
|
18 |
+
│ └── services/
|
19 |
+
│ └── service.yaml
|
20 |
+
├── src/
|
21 |
+
│ ├── __init__.py
|
22 |
+
│ ├── client/
|
23 |
+
│ │ ├── __init__.py
|
24 |
+
│ │ ├── data_handler.py
|
25 |
+
│ │ └── model.py
|
26 |
+
│ ├── server/
|
27 |
+
│ │ ├── __init__.py
|
28 |
+
│ │ ├── aggregator.py
|
29 |
+
│ │ └── coordinator.py
|
30 |
+
│ ├── rag/
|
31 |
+
│ │ ├── __init__.py
|
32 |
+
│ │ ├── retriever.py
|
33 |
+
│ │ └── generator.py
|
34 |
+
│ ├── models/
|
35 |
+
│ │ ├── __init__.py
|
36 |
+
│ │ ├── vae.py
|
37 |
+
│ │ └── gan.py
|
38 |
+
│ └── utils/
|
39 |
+
│ ├── __init__.py
|
40 |
+
│ ├── privacy.py
|
41 |
+
│ └── metrics.py
|
42 |
+
├── tests/
|
43 |
+
│ ├── __init__.py
|
44 |
+
│ ├── test_client.py
|
45 |
+
│ ├── test_server.py
|
46 |
+
│ └── test_rag.py
|
47 |
+
├── docs/
|
48 |
+
│ ├── api/
|
49 |
+
│ ├── guides/
|
50 |
+
│ └── index.md
|
51 |
+
├── notebooks/
|
52 |
+
│ ├── data_exploration.ipynb
|
53 |
+
│ └── model_evaluation.ipynb
|
54 |
+
├── config/
|
55 |
+
│ ├── client_config.yaml
|
56 |
+
│ └── server_config.yaml
|
57 |
+
├── requirements.txt
|
58 |
+
├── setup.py
|
59 |
+
└── README.md
|
60 |
+
```
|
docker/Dockerfile.client
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
FROM python:3.8-slim
|
2 |
+
|
3 |
+
WORKDIR /app
|
4 |
+
|
5 |
+
# Install system dependencies
|
6 |
+
RUN apt-get update && \
|
7 |
+
apt-get install -y --no-install-recommends \
|
8 |
+
build-essential \
|
9 |
+
&& rm -rf /var/lib/apt/lists/*
|
10 |
+
|
11 |
+
# Copy requirements and install Python dependencies
|
12 |
+
COPY requirements.txt .
|
13 |
+
RUN pip install --no-cache-dir -r requirements.txt
|
14 |
+
|
15 |
+
# Copy source code
|
16 |
+
COPY src/ src/
|
17 |
+
COPY config/ config/
|
18 |
+
|
19 |
+
# Set environment variables
|
20 |
+
ENV PYTHONPATH=/app
|
21 |
+
ENV CONFIG_PATH=/app/config/client_config.yaml
|
22 |
+
|
23 |
+
# Run the client
|
24 |
+
CMD ["python", "-m", "src.client.model"]
|
docker/Dockerfile.server
ADDED
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
FROM python:3.8-slim
|
2 |
+
|
3 |
+
WORKDIR /app
|
4 |
+
|
5 |
+
# Install system dependencies
|
6 |
+
RUN apt-get update && \
|
7 |
+
apt-get install -y --no-install-recommends \
|
8 |
+
build-essential \
|
9 |
+
&& rm -rf /var/lib/apt/lists/*
|
10 |
+
|
11 |
+
# Copy requirements and install Python dependencies
|
12 |
+
COPY requirements.txt .
|
13 |
+
RUN pip install --no-cache-dir -r requirements.txt
|
14 |
+
|
15 |
+
# Copy source code
|
16 |
+
COPY src/ src/
|
17 |
+
COPY config/ config/
|
18 |
+
|
19 |
+
# Set environment variables
|
20 |
+
ENV PYTHONPATH=/app
|
21 |
+
ENV CONFIG_PATH=/app/config/server_config.yaml
|
22 |
+
|
23 |
+
# Expose ports for API and monitoring
|
24 |
+
EXPOSE 8000 8001
|
25 |
+
|
26 |
+
# Run the server
|
27 |
+
CMD ["python", "-m", "src.server.coordinator"]
|
docker/docker-compose.yml
ADDED
@@ -0,0 +1,40 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
version: '3.8'
|
2 |
+
|
3 |
+
services:
|
4 |
+
fl-server:
|
5 |
+
build:
|
6 |
+
context: ..
|
7 |
+
dockerfile: docker/Dockerfile.server
|
8 |
+
ports:
|
9 |
+
- "8000:8000"
|
10 |
+
- "8001:8001"
|
11 |
+
volumes:
|
12 |
+
- ../config:/app/config
|
13 |
+
environment:
|
14 |
+
- PYTHONPATH=/app
|
15 |
+
- CONFIG_PATH=/app/config/server_config.yaml
|
16 |
+
networks:
|
17 |
+
- fl-network
|
18 |
+
|
19 |
+
fl-client:
|
20 |
+
build:
|
21 |
+
context: ..
|
22 |
+
dockerfile: docker/Dockerfile.client
|
23 |
+
depends_on:
|
24 |
+
- fl-server
|
25 |
+
volumes:
|
26 |
+
- ../config:/app/config
|
27 |
+
environment:
|
28 |
+
- PYTHONPATH=/app
|
29 |
+
- CONFIG_PATH=/app/config/client_config.yaml
|
30 |
+
- SERVER_HOST=fl-server
|
31 |
+
- SERVER_PORT=8000
|
32 |
+
networks:
|
33 |
+
- fl-network
|
34 |
+
deploy:
|
35 |
+
replicas: 3
|
36 |
+
|
37 |
+
networks:
|
38 |
+
fl-network:
|
39 |
+
driver: bridge
|
40 |
+
|
docs/index.md
ADDED
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Federated Learning for Privacy-Preserving Financial Data Generation
|
2 |
+
|
3 |
+
## Overview
|
4 |
+
This documentation covers the implementation and usage of a federated learning system for generating synthetic financial data with privacy preservation using RAG (Retrieval-Augmented Generation).
|
5 |
+
|
6 |
+
## Quick Start
|
7 |
+
- [Installation Guide](guides/installation.md)
|
8 |
+
- [Usage Guide](guides/usage.md)
|
9 |
+
- [API Reference](api/index.md)
|
10 |
+
- [Project Planning](guides/planning.md)
|
11 |
+
|
12 |
+
## Architecture
|
13 |
+
The system consists of three main components:
|
14 |
+
1. Federated Learning Framework
|
15 |
+
2. Privacy-Preserving Data Generation
|
16 |
+
3. RAG Integration
|
17 |
+
|
18 |
+
## Components
|
19 |
+
- Client Implementation
|
20 |
+
- Server Coordination
|
21 |
+
- RAG System
|
22 |
+
- Privacy Management
|
23 |
+
- Data Handling
|
24 |
+
|
25 |
+
## Contributing
|
26 |
+
Please read our [Contributing Guidelines](guides/contributing.md) for details on submitting pull requests.
|
27 |
+
|
28 |
+
## License
|
29 |
+
This project is licensed under the MIT License - see the LICENSE file for details.
|
30 |
+
|
requirements.txt
ADDED
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Core ML frameworks
|
2 |
+
tensorflow>=2.6.0
|
3 |
+
tensorflow-federated
|
4 |
+
torch>=1.9.0
|
5 |
+
transformers
|
6 |
+
|
7 |
+
# Data processing
|
8 |
+
pandas>=1.3.0
|
9 |
+
numpy>=1.19.0
|
10 |
+
scikit-learn
|
11 |
+
|
12 |
+
# RAG components
|
13 |
+
elasticsearch
|
14 |
+
faiss-cpu
|
15 |
+
|
16 |
+
# Privacy and security
|
17 |
+
tensorflow-privacy
|
18 |
+
pysyft
|
19 |
+
|
20 |
+
# API and web
|
21 |
+
flask
|
22 |
+
fastapi
|
23 |
+
uvicorn
|
24 |
+
|
25 |
+
# Testing and development
|
26 |
+
pytest
|
27 |
+
black
|
28 |
+
flake8
|
29 |
+
isort
|
30 |
+
|
31 |
+
# Documentation
|
32 |
+
sphinx
|
33 |
+
sphinx-rtd-theme
|
34 |
+
|
35 |
+
# Additional requirements
|
36 |
+
pyyaml>=5.4.1
|
src/__init__.py
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
"""__init__.py module."""
|
2 |
+
|
src/client/__init__.py
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
"""__init__.py module."""
|
2 |
+
|
src/models/__init__.py
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
"""__init__.py module."""
|
2 |
+
|
src/rag/__init__.py
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
"""__init__.py module."""
|
2 |
+
|
src/server/__init__.py
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
"""__init__.py module."""
|
2 |
+
|
src/utils/__init__.py
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
"""__init__.py module."""
|
2 |
+
|
tests/__init__.py
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
"""__init__.py module."""
|
2 |
+
|