malwareDetector 0.1.12__py3-none-any.whl → 0.1.13__py3-none-any.whl

malwareDetector/config.py

@@ -39,11 +39,8 @@ class PathConfig(BaseModel, extra=Extra.allow):
 
 class ModelConfig(BaseModel, extra=Extra.allow):
     '''
-    The `ModelConfig` class stores the
-    hyperparameters of model for the detector,
-    which are obtained from `parameter_parser()`.
-    If no additional settings are provided,
-    default values will be used.
+    The `ModelConfig` class is designed to store the
+    parameters and hyperparameters for the detector model.
 
     If you wish to create a customized parameter,
     there has two methods to do it.
@@ -51,15 +48,6 @@ class ModelConfig(BaseModel, extra=Extra.allow):
     method of the `ModelConfig` class.
     2. directly set the parameter in the `config.json` file.
     '''
-    modelName: str = DEFAULT_MODEL_NAME
-    dimensions: int = DEFAULT_DIMENSIONS
-    workers: int = DEFAULT_WORKERS
-    epochs: int = DEFAULT_EPOCHS
-    min_count: int = DEFAULT_MIN_COUNT
-    wl_iterations: int = DEFAULT_WL_ITERATIONS
-    learning_rate: float = DEFAULT_LEARNING_RATE
-    down_sampling: float = DEFAULT_DOWN_SAMPLING
-
     def set_param(self, __name: str, __value: Any):
         if hasattr(self, __name):
             raise AttributeError(f"Parameter {__name} already exists.")
@@ -192,62 +180,6 @@ def parameter_parser(config: Config) -> Config:
                         default=DEFAULT_CONFIG_PATH,
                         help="Configuration file path.")
 
-    parser.add_argument("--dimensions",
-                        dest="dimensions",
-                        type=int,
-                        default=DEFAULT_DIMENSIONS,
-                        help="Number of dimensions. Default is 128."
-                        )
-
-    parser.add_argument("--workers",
-                        dest="workers",
-                        type=int,
-                        default=DEFAULT_WORKERS,
-                        help="Number of workers. Default is 4."
-                        )
-
-    parser.add_argument("--epochs",
-                        dest="epochs",
-                        type=int,
-                        default=DEFAULT_EPOCHS,
-                        help="Number of epochs. Default is 10."
-                        )
-
-    parser.add_argument("--min-count",
-                        dest="min_count",
-                        type=int,
-                        default=DEFAULT_MIN_COUNT,
-                        help="Minimal structural feature count. Default is 5."
-                        )
-
-    parser.add_argument("--wl-iterations",
-                        dest="wl_iterations",
-                        type=int,
-                        default=DEFAULT_WL_ITERATIONS,
-                        help="Number of Weisfeiler-Lehman iterations. Default is 2."
-                        )
-
-    parser.add_argument("--learning-rate",
-                        dest="learning_rate",
-                        type=float,
-                        default=DEFAULT_LEARNING_RATE,
-                        help="Initial learning rate. Default is 0.025."
-                        )
-
-    parser.add_argument("--down-sampling",
-                        dest="down_sampling",
-                        type=float,
-                        default=DEFAULT_DOWN_SAMPLING,
-                        help="Down sampling rate of features. Default is 0.0001."
-                        )
-
-    parser.add_argument('--model','-m',
-                        dest="model",
-                        nargs='?',
-                        default=DEFAULT_MODEL_NAME,
-                        help='Select the model(KNN, LR, MLP, RF, SVM).'
-                        )
-
     parser.add_argument('--classify', '-f',
                         dest="classify",
                         action='store_true',
@@ -259,14 +191,6 @@ def parameter_parser(config: Config) -> Config:
     config.path.input = args.input_path
     config.path.output = args.output_path
     config.path.config = args.config_path
-    config.model.dimensions = args.dimensions
-    config.model.workers = args.workers
-    config.model.epochs = args.epochs
-    config.model.min_count = args.min_count
-    config.model.wl_iterations = args.wl_iterations
-    config.model.learning_rate = args.learning_rate
-    config.model.down_sampling = args.down_sampling
-    config.model.modelName = args.model
     config.classify = args.classify
 
     return config

malwareDetector/const.py

@@ -2,17 +2,9 @@
 Save the variables of the system's default constants.
 '''
 
-DEFAULT_INPUT_PATH="./Dataset/malware"
-DEFAULT_OUT_PATH="./Feature/feature.csv"
+DEFAULT_INPUT_PATH="./Dataset/program"
+DEFAULT_OUT_PATH="./Predict/predict.json"
 DEFAULT_CONFIG_PATH="./config.json"
-DEFAULT_MODEL_NAME="SVM"
-DEFAULT_DIMENSIONS=128
-DEFAULT_WORKERS=4
-DEFAULT_EPOCHS=50
-DEFAULT_MIN_COUNT=1
-DEFAULT_WL_ITERATIONS=2
-DEFAULT_LEARNING_RATE=0.025
-DEFAULT_DOWN_SAMPLING=0.0001
 DEFAULT_CLASSIFY=False
 
 DATASET_DIR="./Dataset/"

malwareDetector/detector.py

@@ -1,15 +1,19 @@
 import os
 import numpy as np
+
 from typing import Any
 from .config import read_config
 from .utils import platform_info
+from .const import DEFAULT_CONFIG_PATH
 
 class detector(object):
-    def __init__(self) -> None:
+    def __init__(self, config_path=None) -> None:
         '''
         read config.json by `read_config` function in `config.py`
         '''
-        self.config = read_config()
+        if config_path is None:
+            config_path = DEFAULT_CONFIG_PATH
+        self.config = read_config(config_path)
 
     def extractFeature(self) -> Any:
         '''

{malwareDetector-0.1.12.dist-info → malwareDetector-0.1.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: malwareDetector
-Version: 0.1.12
+Version: 0.1.13
 Summary: Malware detector specification for NTUST isLab
 Author: PO-LIN LAI
 Author-email: bolin8017@gmail.com
@@ -11,49 +11,45 @@ License-File: LICENCE.txt
 Requires-Dist: numpy
 Requires-Dist: pydantic
 
-malwareDetector
-===============
+# malwareDetector
 
 * Source code: https://github.com/louiskyee/malwareDetector.git
 * Wiki: https://github.com/louiskyee/malwareDetector/wiki
 * PyPI: https://pypi.org/project/malwareDetector/
 
-Description
------------
+## Description
 
 This is a malware detector specification for NTUST isLab.
 The `malwareDetector` is a base class designed for malicious software detection. It enables straightforward utilization of Python's inheritance feature. By inheriting from `malwareDetector` and implementing the required functions, you can achieve your specific goals. Additionally, it offers convenient configuration management. For more detailed instructions, please refer to the [GitHub Wiki](https://github.com/louiskyee/malwareDetector/wiki).
 
-Requirements
-------------
-Tool | Version |Source |
-|---|---|---|
-| Python | `>= 3.10` | https://www.python.org/downloads |
+## Requirements
 
-Installation
-------------
+Tool | Version | Source
+-----|---------|-------
+Python | `>= 3.10` | https://www.python.org/downloads
+
+## Installation
 
 Use the package manager [pip](https://pip.pypa.io/en/stable/) to install `malwareDetector`.
 * Example: `pip install malwareDetector`
 
-Usage
------
+## Usage
 
-### import
-* import class `detector` from `malwareDetector.detector`
-    ```python=
+### Import
+* Import class `detector` from `malwareDetector.detector`
+    ```python
     from malwareDetector.detector import detector
     ```
 
-### Examples:
-```python=
-from malwareDetector.detector import detector
-from typing import Any
+### Example:
+```python
 import numpy as np
+from typing import Any
+from malwareDetector.detector import detector
 
 class subDetector(detector):
-    def __init__(self) -> None:
-        super().__init__()
+    def __init__(self, config_path=None) -> None:
+        super().__init__(config_path)
 
     def extractFeature(self) -> Any:
         return 'This is the implementation of the extractFeature function from the derived class.'
@@ -61,9 +57,22 @@ class subDetector(detector):
     def vectorize(self) -> np.array:
         return 'This is the implementation of the vectorize function from the derived class.'
 
-    def model(self) -> Any:
+    def model(self, training: bool = True) -> Any:
         return 'This is the implementation of the model function from the derived class.'
 
     def predict(self) -> np.array:
         return 'This is the implementation of the predict function from the derived class.'
 ```
+
+## Configuration
+
+The `malwareDetector` uses a configuration system that can be customized through a JSON file or command-line arguments. The default configuration file is `config.json` in the current directory, but you can specify a custom path when initializing the detector.
+
+### Key Configuration Classes:
+
+- `Config`: Stores all external settings for the detector.
+- `PathConfig`: Manages input and output file paths.
+- `FolderConfig`: Handles folder configurations for data storage.
+- `ModelConfig`: Stores model-specific parameters and hyperparameters.
+
+For detailed information on configuration options and usage, please refer to the [GitHub Wiki](https://github.com/louiskyee/malwareDetector/wiki).

malwareDetector-0.1.13.dist-info/RECORD

@@ -0,0 +1,10 @@
+malwareDetector/__init__.py,sha256=x9kkfeWTUR0g6RQkE13V2sZhY2DSVD3KqzcxOqlNjtA,768
+malwareDetector/config.py,sha256=EOAFPqLyKMFvg8fEc9utpadCqdTeMGYL-312E9hF6Hs,9397
+malwareDetector/const.py,sha256=JJCGc9eniIpyrV5YGtCqjEPbeQYkJw3jTiaJb_Dt9EE,341
+malwareDetector/detector.py,sha256=nyfnLjoxr0l3eL-HyuWJ-6EoBn28_lYEYwStsYH7dog,2358
+malwareDetector/utils.py,sha256=kjEX-A-FuwjGwxZ-2JP489NaaeC7HwvdMrDrfPk4pYE,371
+malwareDetector-0.1.13.dist-info/LICENCE.txt,sha256=2XPCaZqZ-jgHh7e1DKa87JUeuOB6DC0jaZonmjDeILM,1088
+malwareDetector-0.1.13.dist-info/METADATA,sha256=ZokOssLklL5ZTNNtJNTjkb6_Gb_gvHts2mgAZojZ91I,3014
+malwareDetector-0.1.13.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+malwareDetector-0.1.13.dist-info/top_level.txt,sha256=wRXSanQD5XDXRYp3lPh1SjltOo6rpC5jktmR69tqIQo,16
+malwareDetector-0.1.13.dist-info/RECORD,,

{malwareDetector-0.1.12.dist-info → malwareDetector-0.1.13.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.38.4)
+Generator: bdist_wheel (0.43.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

malwareDetector-0.1.12.dist-info/RECORD

@@ -1,10 +0,0 @@
-malwareDetector/__init__.py,sha256=x9kkfeWTUR0g6RQkE13V2sZhY2DSVD3KqzcxOqlNjtA,768
-malwareDetector/config.py,sha256=a00Whtx_2vdaJQH7luJECEMOIlyfaMLHbDHyfncMo-k,12430
-malwareDetector/const.py,sha256=8TKTqjS70sU3cBFSwFUaWmoLiguRbi0ul63xswtA_Ho,533
-malwareDetector/detector.py,sha256=IeLfmOoMp20s-io5w8igMc5bJdXt8ir-EarHKBo3C_U,2207
-malwareDetector/utils.py,sha256=kjEX-A-FuwjGwxZ-2JP489NaaeC7HwvdMrDrfPk4pYE,371
-malwareDetector-0.1.12.dist-info/LICENCE.txt,sha256=2XPCaZqZ-jgHh7e1DKa87JUeuOB6DC0jaZonmjDeILM,1088
-malwareDetector-0.1.12.dist-info/METADATA,sha256=uE3uqGROsWh6Yy9UDTWB6EaGsropygdRX7a38Z3Svuw,2287
-malwareDetector-0.1.12.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-malwareDetector-0.1.12.dist-info/top_level.txt,sha256=wRXSanQD5XDXRYp3lPh1SjltOo6rpC5jktmR69tqIQo,16
-malwareDetector-0.1.12.dist-info/RECORD,,