Doctorado-ML/stree
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/STree.git synced 2025-08-15 15:36:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update notebooks and readme with cosmetic changes

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2020-05-20 18:11:57 +02:00
parent 316f84cc63
commit a4595f5815
5 changed files with 188 additions and 182 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
trees/Snode_graph.py
View File

@@ -31,12 +31,12 @@ class Snode_graph(Snode):
return False
def plot_hyperplane(self):
# get the splitting hyperplane
def hyperplane(x, y): return (-self._interceptor - self._vector[0][0] * x
- self._vector[0][1] * y) / self._vector[0][2]
fig = plt.figure(figsize=self._plot_size)
ax = fig.add_subplot(1, 1, 1, projection='3d')
if not self._is_pure():
# get the splitting hyperplane
def hyperplane(x, y): return (-self._interceptor - self._vector[0][0] * x
- self._vector[0][1] * y) / self._vector[0][2]
# Can't plot hyperplane of leaves with one label because it hasn't classiffier
tmpx = np.linspace(self._X[:, 0].min(), self._X[:, 0].max())
tmpy = np.linspace(self._X[:, 1].min(), self._X[:, 1].max())