Black ...

slickml/classification.py

@@ -311,7 +311,7 @@ def plot_feature_importance(
  markerfacecolor=None,
  markeredgewidth=None,
  fontsize=None,
- save_path=None
+ save_path=None,
  ):
 
  """Function to plot XGBoost feature importance.
@@ -348,10 +348,10 @@ def plot_feature_importance(
 
  fontsize: int or float, optional, (default=12)
  Fontsize for xlabel and ylabel, and ticks parameters
- 
+
  save_path: str, optional (default=None)
  The full or relative path to save the plot including the image format.
- For example "myplot.png" or "../../myplot.pdf" 
+ For example "myplot.png" or "../../myplot.pdf"
  """
 
  plot_xgb_feature_importance(
@@ -364,7 +364,7 @@ def plot_feature_importance(
  markerfacecolor=markerfacecolor,
  markeredgewidth=markeredgewidth,
  fontsize=fontsize,
- save_path=save_path
+ save_path=save_path,
  )
 
  def plot_shap_summary(
@@ -383,7 +383,7 @@ def plot_shap_summary(
  class_names=None,
  class_inds=None,
  color_bar_label=None,
- save_path=None
+ save_path=None,
  ):
  """Function to plot shap summary plot.
  This function is a helper function to plot the shap summary plot
@@ -440,10 +440,10 @@ def plot_shap_summary(
 
  color_bar_label: str, optional, (default="Feature Value")
  Label for color bar
- 
+
  save_path: str, optional (default=None)
  The full or relative path to save the plot including the image format.
- For example "myplot.png" or "../../myplot.pdf" 
+ For example "myplot.png" or "../../myplot.pdf"
  """
 
  # define tree explainer
@@ -477,7 +477,7 @@ def plot_shap_summary(
  class_names=class_names,
  class_inds=class_inds,
  color_bar_label=color_bar_label,
- save_path=save_path
+ save_path=save_path,
  )
 
  def plot_shap_waterfall(
@@ -495,7 +495,7 @@ def plot_shap_waterfall(
  max_display=None,
  title=None,
  fontsize=None,
- save_path=None
+ save_path=None,
  ):
  """Function to plot shap waterfall plot.
  This function is a helper function to plot the shap waterfall plot
@@ -549,10 +549,10 @@ def plot_shap_waterfall(
 
  fontsize: int or float, optional, (default=12)
  Fontsize for xlabel and ylabel, and ticks parameters
- 
+
  save_path: str, optional (default=None)
  The full or relative path to save the plot including the image format.
- For example "myplot.png" or "../../myplot.pdf" 
+ For example "myplot.png" or "../../myplot.pdf"
  """
 
  # define tree explainer
@@ -585,7 +585,7 @@ def plot_shap_waterfall(
  max_display=max_display,
  title=title,
  fontsize=fontsize,
- save_path=save_path
+ save_path=save_path,
  )
 
  def _dtrain(self, X_train, y_train):
@@ -1044,7 +1044,7 @@ def plot_cv_results(
  train_std_color=None,
  test_color=None,
  test_std_color=None,
- save_path=None
+ save_path=None,
  ):
  """
  Function to plot the results of xgboost.cv() process and evolution
@@ -1079,10 +1079,10 @@ def plot_cv_results(
 
  test_std_color: str, optional, (default="#D0AAF3")
  Color of the edge color of the testing std bars
- 
+
  save_path: str, optional (default=None)
  The full or relative path to save the plot including the image format.
- For example "myplot.png" or "../../myplot.pdf" 
+ For example "myplot.png" or "../../myplot.pdf"
  """
 
  plot_xgb_cv_results(
@@ -1095,7 +1095,7 @@ def plot_cv_results(
  train_std_color=train_std_color,
  test_color=test_color,
  test_std_color=test_std_color,
- save_path=save_path
+ save_path=save_path,
  )
 
  def _cv(self):

slickml/feature_selection.py

@@ -572,7 +572,7 @@ def plot_frequency(
  markerfacecolor=None,
  markeredgewidth=None,
  fontsize=None,
- save_path=None
+ save_path=None,
  ):
 
  """Function to plot selected features frequency.
@@ -625,11 +625,17 @@ def plot_frequency(
  markerfacecolor=markerfacecolor,
  markeredgewidth=markeredgewidth,
  fontsize=fontsize,
- save_path=save_path
+ save_path=save_path,
  )
 
  def plot_cv_results(
- self, figsize=None, int_color=None, ext_color=None, sharex=False, sharey=False, save_path=None
+ self,
+ figsize=None,
+ int_color=None,
+ ext_color=None,
+ sharex=False,
+ sharey=False,
+ save_path=None,
  ):
  """Function to plot the cross-validation results of
  XGBoostFeatureSelector. It visualizes the internal
@@ -656,10 +662,10 @@ def plot_cv_results(
 
  sharey: bool, optional, (default=False)
  Flag to share "Y" axis for each row of subplots
- 
+
  save_path: str, optional (default=None)
  The full or relative path to save the plot including the image format.
- For example "myplot.png" or "../../myplot.pdf" 
+ For example "myplot.png" or "../../myplot.pdf"
 
  kwargs: dict
  Plotting object plotting_cv_
@@ -668,10 +674,11 @@ def plot_cv_results(
  plot_xfs_cv_results(
  figsize=figsize,
  int_color=int_color,
- ext_color=ext_color, 
- sharex=sharex, sharey=sharey,
+ ext_color=ext_color,
+ sharex=sharex,
+ sharey=sharey,
  save_path=save_path,
- **self.plotting_cv_
+ **self.plotting_cv_,
  )
 
  def get_xgb_params(self):

slickml/metrics.py

@@ -545,15 +545,15 @@ def plot(self, figsize=None, save_path=None):
  ----------
  figsize: tuple, optional, (default=(12, 12))
  Figure size
- 
+
  save_path: str, optional (default=None)
  The full or relative path to save the plot including the image format.
- For example "myplot.png" or "../../myplot.pdf" 
+ For example "myplot.png" or "../../myplot.pdf"
  """
 
- plot_binary_classification_metrics(figsize=figsize,
-  save_path=save_path,
-  **self.plotting_dict_)
+ plot_binary_classification_metrics(
+ figsize=figsize, save_path=save_path, **self.plotting_dict_
+ )
 
 
 class RegressionMetrics:
@@ -910,10 +910,12 @@ def plot(self, figsize=None, save_path=None):
  ----------
  figsize: tuple, optional, (default=(12, 12))
  Figure size
- 
+
  save_path: str, optional (default=None)
  The full or relative path to save the plot including the image format.
- For example "myplot.png" or "../../myplot.pdf" 
+ For example "myplot.png" or "../../myplot.pdf"
  """
 
- plot_regression_metrics(figsize=figsize, save_path=save_path, **self.plotting_dict_)
+ plot_regression_metrics(
+ figsize=figsize, save_path=save_path, **self.plotting_dict_
+ )