Restore row cout boxen plot updated

This commit is contained in:
Sudipta Basak 2024-06-04 21:28:22 +00:00
parent 86e9a78f34
commit bfd75feaa9
No known key found for this signature in database

View File

@ -274,10 +274,22 @@ class Visualizor:
conf["title"] = "Boxen Plot of Restore Row Count by Nodes"
conf["xlabel"] = "Restore Row Count"
conf["ylabel"] = "Nodes"
n1 = int(result.numberNodes * result.class1ratio)
data = [result.restoreRowCount[1: n1], result.restoreRowCount[n1+1: ]]
data = []
nodeClasses, nodeRanges = self.__getNodeRanges(result.shape)
_start = 1
for _range in nodeRanges:
data.append(result.restoreRowCount[_start: _range])
_start = _range
_values, _categories = [], []
for _d, _nc in zip(data, nodeClasses):
_values += _d
_categories += [f'Class {_nc}'] * len(_d)
data = pd.DataFrame({
'values': _values,
'category': _categories
})
plt.figure(figsize=(8, 6))
sns.boxenplot(data=data, width=0.8)
sns.boxenplot(x='category', y='values', data=data, palette="Set2", ax=plt.gca(), width=0.8)
plt.xlabel(conf["xlabel"], fontsize=12)
plt.ylabel(conf["ylabel"], fontsize=12)
plt.title(conf["title"], fontsize=14)