Add setup.py

mini-batch · mini-batch · commit 277b68261566 · 2023-07-21T22:43:01.000+01:00
diff --git a/nn_generalisation/landscape_analysis/__init__.py b/nn_generalisation/landscape_analysis/__init__.py
diff --git a/nn_generalisation/landscape_analysis/slope.py b/nn_generalisation/landscape_analysis/slope.py
@@ -4,8 +4,6 @@
 
 import torch
 import torch.nn.functional as F
-import matplotlib.pyplot as plt
-import numpy as np
 
 def get_jacobian_avg(model : DenseNN,
                      exp : Experiment) -> float:
diff --git a/setup.py b/setup.py
@@ -0,0 +1,6 @@
+from setuptools import setup
+
+setup(name='nn_generalisation',
+   version='0.1',
+   author='David Batchelder',
+   packages=['nn_generalisation'])
