Skip to content

Commit a464e0d

Browse files
Rename generator and trainer scripts; update README for directory structure and file naming consistency
1 parent 081af14 commit a464e0d

File tree

3 files changed

+9
-13
lines changed

3 files changed

+9
-13
lines changed
File renamed without changes.

README.md

Lines changed: 9 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -128,20 +128,16 @@ as well as previously trained models for you to test out
128128
The repository is located [here](https://github.com/DefinetlyNotAI/VulnScan_Data).
129129

130130
The repository contains the following directories:
131-
- `Archived Models`: Contains the previously trained models. Is organized by the model type then version.
131+
- `cache`: Contains all training data generated by [`Generator.py`](Generator.py).
132132
- `NN features`: Contains information about the model `.3n3` and the vectorizer used. Information include:
133133
- `Documentation_Study_Network.md`: A markdown file that contains more info.
134-
- `Neural Network Nodes Graph.gexf`: A Gephi file that contains the model nodes and edges.
135-
- `Nodes and edges (GEPHI).csv`: A CSV file that contains the model nodes and edges.
136-
- `Statistics`: Directories made by Gephi, containing the statistics of the model nodes and edges.
134+
- `Neural_Network_Nodes_Graph.gexf`: A Gephi file that contains the model nodes and edges.
137135
- `Feature_Importance.svg`: A SVG file that contains the feature importance of the model.
138136
- `Loss_Landscape_3D.html`: A HTML file that contains the 3D loss landscape of the model.
139-
- `Model Accuracy Over Epochs.png` and `Model Loss Over Epochs.png`: PNG files that contain the model accuracy and loss over epochs.
140-
- `Model state dictionary.txt`: A text file that contains the model state dictionary.
141-
- `Model Summary.txt`: A text file that contains the model summary.
142-
- `Model Visualization.png`: A PNG file that contains the model visualization.
143-
- `Top_90_Features.svg`: A SVG file that contains the top 90 features of the model.
144-
- `Vectorizer features.txt`: A text file that contains the vectorizer features.
145-
- `Visualize Activation.png`: A PNG file that contains the visualization of the model activation.
146-
- `Visualize t-SNE.png`: A PNG file that contains the visualization of the model t-SNE.
147-
- `Weight Distribution.png`: A PNG file that contains the weight distribution of the model.
137+
- `Model_State_Dict.txt`: A text file that contains the model state dictionary.
138+
- `Model_Summary.txt`: A text file that contains the model summary.
139+
- `Model_Visualization.png`: A PNG file that contains the model visualization.
140+
- `Visualize_Activation.png`: A PNG file that contains the visualization of the model activation.
141+
- `Visualize_tSNE.png`: A PNG file that contains the visualization of the model t-SNE with the default training test embeds.
142+
- `Visualize_tSNE_custom.png`: A PNG file that contains the visualization of the model t-SNE with real world training examples (only 100).
143+
- `Weight_Distribution.png`: A PNG file that contains the weight distribution of the model.
File renamed without changes.

0 commit comments

Comments
 (0)