あるコードの入出力などの関係性を図示したい

LLMに図示してもらう

mermaid書式

それっぽい見た目にはなった

本当にあっているかは謎

code: mermaid

%% data_generator,model,train,evalの関係性を図示

flowchart TD

%% ── サブグラフ定義 ─────────────────────────────────────

subgraph Data_Generation

DG -->|"generate_multilabel_dataset(df, audio_dir,...) → (X: np.array(N,n_mels,T), y: np.array(N,C))"| DSDataset

DS -->|"save_sample(..., fmt='npz') → dataset.npz"| NPZfile: dataset.npz

end

subgraph Training

NPZ -->|"np.load('dataset.npz') → X: (N,n_mels,T), y: (N,C)"| T

T -->|uses get_model' from| Mmodel.py

T -->|"train & save → model.pth"| Wfile: model.pth

end

subgraph Evaluation

NPZ -->|"np.load('dataset.npz')"| E

W -->|"torch.load('model.pth')"| E

E -->|"infer + threshold → preds_bin: (N,C)"| RResults

end

%% ── モデル定義依存 ────────────────────────────────────

M -.-> T

M -.-> E