Run ruff
This commit is contained in:
@@ -586,8 +586,10 @@
|
||||
}
|
||||
],
|
||||
"source": [
|
||||
"data = pd.DataFrame(load_breast_cancer().data, columns=load_breast_cancer().feature_names)\n",
|
||||
"data['target'] = load_breast_cancer().target\n",
|
||||
"data = pd.DataFrame(\n",
|
||||
" load_breast_cancer().data, columns=load_breast_cancer().feature_names\n",
|
||||
")\n",
|
||||
"data[\"target\"] = load_breast_cancer().target\n",
|
||||
"data.head()"
|
||||
]
|
||||
},
|
||||
@@ -619,7 +621,7 @@
|
||||
}
|
||||
],
|
||||
"source": [
|
||||
"ax = sns.countplot(x='target', data=data)\n",
|
||||
"ax = sns.countplot(x=\"target\", data=data)\n",
|
||||
"ax.set_title(\"Type of Tumors (Malignant/Benign)\")"
|
||||
]
|
||||
},
|
||||
@@ -683,14 +685,16 @@
|
||||
}
|
||||
],
|
||||
"source": [
|
||||
"df = data.melt(id_vars='target', var_name='feature', value_name='value')\n",
|
||||
"df = data.melt(id_vars=\"target\", var_name=\"feature\", value_name=\"value\")\n",
|
||||
"\n",
|
||||
"GROUP_SIZE = 10\n",
|
||||
"for start in range(0, len(load_breast_cancer().feature_names), GROUP_SIZE):\n",
|
||||
" end = start + GROUP_SIZE\n",
|
||||
" subset = df[df['feature'].isin(load_breast_cancer().feature_names[start:end])]\n",
|
||||
" sns.violinplot(x='feature', y='value', hue='target', data=subset, split=True, inner='quart')\n",
|
||||
" plt.xticks(rotation=45, ha='right')\n",
|
||||
" subset = df[df[\"feature\"].isin(load_breast_cancer().feature_names[start:end])]\n",
|
||||
" sns.violinplot(\n",
|
||||
" x=\"feature\", y=\"value\", hue=\"target\", data=subset, split=True, inner=\"quart\"\n",
|
||||
" )\n",
|
||||
" plt.xticks(rotation=45, ha=\"right\")\n",
|
||||
" plt.tight_layout()\n",
|
||||
" plt.savefig(f\"violin_plot_features_{start}.pdf\")\n",
|
||||
"\n",
|
||||
|
||||
Reference in New Issue
Block a user