Add machine learning feature extraction for PE files
- Implemented MachineLearning class with ExtractFeatures method - Updated project files to include new machine learning source and header files - Modified main executable to call feature extraction - Updated VSCode settings to include additional C++ headers - Commented out previous file dumping code in main function
This commit is contained in:
@@ -46,7 +46,9 @@ int main() {
|
||||
if (peBuffer) {
|
||||
printf("peBuffer: %p\n", peBuffer.get());
|
||||
printf("peSize: %d\n", peSize);
|
||||
peconv::dump_to_file("z:\\dumped_main.exe", peBuffer.get(), peSize);
|
||||
// peconv::dump_to_file("z:\\dumped_main.exe", peBuffer.get(), peSize);
|
||||
MachineLearning ml;
|
||||
ml.ExtractFeatures(peBuffer.get(), peSize, "z:\\features.txt");
|
||||
}
|
||||
peBuffer.release();
|
||||
system("pause");
|
||||
|
||||
Reference in New Issue
Block a user