예제 #1
0
    def test2(self):
        useless = ['RegionID', 'resulttime', 'CellID', 'PRB_DL_Used_Rate']
        analyzer.drop_unnecessary_columns(useless)
        arguments = {"C": 0.3}
        analyzer.create_model(ModelType.logregression, arguments,
                              analyzer.data,
                              analyzer.data['cnt_averload_cell'])

        print("Cross variation score:\n" + str(analyzer.models[0].cv_score))
        # print("Feature importances:\n" + str(analyzer.models[0].model.feature_importances_))
        print(analyzer.models[0].feature_names)
        Visualizer.draw_class_scatter(analyzer.models[0])
예제 #2
0
    def test3(self):
        classifier = "DL_MCS_64QAM"
        # TODO: redo it, so it doesn't delete the classifier from the dataset, what if you want to reuse it?
        classifier_values = analyzer.data[classifier]
        arguments = {"C": 0.8, "kernel": 'linear'}
        useless = [
            'RegionID', 'resulttime', 'CellID', 'cnt_averload_cell', classifier
        ]
        analyzer.drop_unnecessary_columns(useless)
        analyzer.create_model(ModelType.svc, arguments, analyzer.data,
                              classifier_values)

        print("Cross variation score:\n" + str(analyzer.models[0].cv_score))
        # print("Feature importances:\n" + str(analyzer.models[0].model.feature_importances_))
        print(analyzer.models[0].feature_names)
        Visualizer.draw_class_scatter(analyzer.models[0])