Python train_classifier示例

示例#1

文件： test_machineLearningFramework.py 项目： 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()

示例#2

文件： test_machineLearningFramework.py 项目： 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)

示例#3

文件： test_machineLearningFramework.py 项目： 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)

示例#4

文件： test_machineLearningFramework.py 项目： 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()

示例#5

文件： test_machineLearningFramework.py 项目： marcusCarpenter97/Machine-Learning-for-Malware-Detection

def test_train_classifier_without_classifier():

    ret_val = mlf.train_classifier()

    assert isinstance(ret_val, str)