from FeatureExtractor import FeatureExtractor from FeatureExtractorWitfhDistance import FeatureExtractorWitfhDistance from Classifier import Classifier from ForestClassifier import ForestClassifier if __name__ == "__main__": extractFeatures = False if extractFeatures: featureExtractor = FeatureExtractor( 'pcap_dir/test/icmp_tunnel_pcap', 'feature_csv_file/test_icmp_tunnel_pcap.csv') # featureExtractor = FeatureExtractorWitfhDistance('pcap_dir/file_and_cmd_pcap', 'feature_csv_file/file_and_cmd_features.csv') featureExtractor.extractFeaturesFromIPPairs() else: classifier = Classifier() # classifier = ForestClassifier() classifier.classifier('feature_csv_file/test_icmp_tunnel_pcap.csv')