main.py - cmikke97/Automatic-Malware-Signature-Generation GitHub Wiki
-
import configparser- implements a basic configuration language for Python programs - configparser documentation -
import json- json encoder and decoder - json documentation -
import os- provides a portable way of using operating system dependent functionality - os documentation -
from urllib import parse- standard interface to break Uniform Resource Locator (URL) in components - urllib.parse documentation
-
import baker- easy, powerful access to Python functions from the command line - baker documentation -
import mlflow- open source platform for managing the end-to-end machine learning lifecycle - mlflow documentation -
from logzero import logger- robust and effective logging for Python - logzero documentation -
from mlflow.utils import mlflow_tags- mlflow tags - mlflow documentation
from FreshDatasetBuilder.utils.fresh_dataset_utils import check_files as fresh_check_filesfrom Sorel20mDataset.utils.download_utils import check_files as download_check_filesfrom Sorel20mDataset.utils.preproc_utils import check_files as preproc_check_filesfrom utils.workflow_utils import Hashfrom utils.workflow_utils import get_or_runfrom utils.workflow_utils import run
workflow(base_dir, use_cache, ignore_git) (function, baker command) - Automatic Malware Signature Generation MLflow workflow.
-
base_dir(arg) - Base tool path -
use_cache(arg) - Whether to skip already executed runs (in cache) or not (1/0) (default: 1) -
ignore_git(arg) - Whether to ignore git version or not (1/0) (default: 0)
aloha_workflow(base_dir, use_cache, ignore_git) (function, baker command) - ALOHA MLflow workflow.
-
base_dir(arg) - Base tool path -
use_cache(arg) - Whether to skip already executed runs (in cache) or not (1/0) (default: 1) -
ignore_git(arg) - Whether to ignore git version or not (1/0) (default: 0)
family_classifier_only(base_dir, use_cache, ignore_git) (function, baker command) - Family Classifier Only (with no Trasnfer Learning) workflow.
-
base_dir(arg) - Base tool path -
use_cache(arg) - Whether to skip already executed runs (in cache) or not (1/0) (default: 1) -
ignore_git(arg) - Whether to ignore git version or not (1/0) (default: 0)
contrastive_learning_only(base_dir, use_cache, ignore_git) (function, baker command) - Contrastive Learning Only (with no Transfer Learning) workflow.
-
base_dir(arg) - Base tool path -
use_cache(arg) - Whether to skip already executed runs (in cache) or not (1/0) (default: 1) -
ignore_git(arg) - Whether to ignore git version or not (1/0) (default: 0)
__main__ (main) - Start baker in order to make it possible to run the script and use function names and parameters as the command line interface, using optparse-style options