diff options
Diffstat (limited to 'Assignment 6/packages')
-rw-r--r-- | Assignment 6/packages/decision_boundaries.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/Assignment 6/packages/decision_boundaries.py b/Assignment 6/packages/decision_boundaries.py new file mode 100644 index 0000000..7131ba9 --- /dev/null +++ b/Assignment 6/packages/decision_boundaries.py @@ -0,0 +1,15 @@ +#for a neural network object nw ...
+nw =
+
+figure(1); hold(True)
+delta = 0.05; levels = 100
+a = arange(-1,2,delta)
+b = arange(-1,2,delta)
+A, B = meshgrid(a, b)
+values = np.zeros(A.shape)
+
+for i in range(len(a)):
+ for j in range(len(b)):
+ values[i,j] = nw.sim( np.mat([a[i],b[j]]) )[0,0]
+contour(A, B, values, levels=[.5], colors=['k'], linestyles='dashed')
+contourf(A, B, values, levels=linspace(values.min(),values.max(),levels), cmap=cm.RdBu)
|