| import json
|
| import pandas as pd
|
|
|
| def load_json_to_dataframe(json_file, label):
|
| """Carrega um JSON de c贸digos e adiciona a coluna de r贸tulo (label)."""
|
| with open(json_file, 'r', encoding='utf-8') as f:
|
| data = json.load(f)
|
| df = pd.DataFrame(data)
|
| df['label'] = label
|
| return df
|
|
|
| if __name__ == "__main__":
|
|
|
| safe_json = "safe-code-analyzer/compiled_safe_codes.json"
|
| vuln_json = "exploit-analyzer/compiled_exploits.json"
|
|
|
|
|
| safe_df = load_json_to_dataframe(safe_json, label=0)
|
| vuln_df = load_json_to_dataframe(vuln_json, label=1)
|
|
|
|
|
| dataset_df = pd.concat([safe_df, vuln_df], ignore_index=True)
|
|
|
|
|
| dataset_df.to_csv("code_analysis_dataset.csv", index=False)
|
| print("[SUCCESS] Dataset saved to code_analysis_dataset.csv")
|
|
|
