diff --git a/DeviceClassifier/OneLayer/eval_OneLayer.py b/DeviceClassifier/OneLayer/eval_OneLayer.py index ebd2c6c1..634a884e 100644 --- a/DeviceClassifier/OneLayer/eval_OneLayer.py +++ b/DeviceClassifier/OneLayer/eval_OneLayer.py @@ -160,7 +160,7 @@ def main(self): abnormality = 0 abnormality = eval_pcap( str(pcap), self.conf_labels, self.time_const, label=labels[0], - rnn_size=self.rnn_size, model_type='OneLayer') + rnn_size=self.rnn_size, model_path=load_path, model_type='OneLayer') prev_s = self.common.get_address_info( source_mac, timestamp diff --git a/requirements.txt b/requirements.txt index 9f2a3be5..b409ede3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ numpy==1.16.1 pika==0.13.0 -pytest==4.2.1 +pytest==4.3.0 pytest-cov==2.6.1 -redis==3.1.0 +redis==3.2.0 scikit-learn==0.20.2 tensorflow==1.12.0 diff --git a/setup.py b/setup.py index a8ca9cac..c7510eb4 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,7 @@ packages=['poseidonml'], package_dir={'poseidonml': 'utils'}, package_data={'poseidonml': ['models/*']}, - install_requires=['numpy==1.16.1', 'pika==0.13.0', 'redis==3.1.0', + install_requires=['numpy==1.16.1', 'pika==0.13.0', 'redis==3.2.0', 'scikit-learn==0.20.2', 'tensorflow==1.12.0'], license='Apache License 2.0', author='cglewis',