Python train_classifier Examples

Example #1

File: test_machineLearningFramework.py Project: marcusCarpenter97/Machine-Learning-for-Malware-Detection

def test_train_classifier_with_classifier():

    malware_rate = 0.45
    partition_size = 0.92

    mlf.load_json_data(malware_rate)

    mlf.partition_data(partition_size)

    mlf.select_classifier("Decision Tree")

    mlf.train_classifier()

Example #2

File: test_machineLearningFramework.py Project: marcusCarpenter97/Machine-Learning-for-Malware-Detection

def test_calculate_classifier_accuracy_with_classifier():

    malware_rate = 0.33
    partition_size = 0.33

    mlf.load_json_data(malware_rate)

    mlf.partition_data(partition_size)

    mlf.select_classifier("Decision Tree")

    mlf.train_classifier()

    ret_val = mlf.calculate_classifier_accuracy()

    assert isinstance(ret_val, float)

Example #3

File: test_machineLearningFramework.py Project: marcusCarpenter97/Machine-Learning-for-Malware-Detection

def test_classify_new_data_with_classifier():

    malware_rate = 0.33
    partition_size = 0.33

    mlf.load_json_data(malware_rate)

    mlf.partition_data(partition_size)

    mlf.select_classifier("Decision Tree")

    mlf.train_classifier()

    ret_val = mlf.classify_new_data()

    assert isinstance(ret_val, numpy.ndarray)

Example #4

File: test_machineLearningFramework.py Project: marcusCarpenter97/Machine-Learning-for-Malware-Detection

def test_display_confusion_matrix_pass():

    malware_rate = 0.33
    partition_size = 0.33

    mlf.REPORT.create_new_report_directory()

    mlf.load_json_data(malware_rate)

    mlf.partition_data(partition_size)

    mlf.select_classifier("Decision Tree")

    mlf.train_classifier()

    mlf.classify_new_data()

    mlf.display_confusion_matrix()

Example #5

File: test_machineLearningFramework.py Project: marcusCarpenter97/Machine-Learning-for-Malware-Detection

def test_train_classifier_without_classifier():

    ret_val = mlf.train_classifier()

    assert isinstance(ret_val, str)