self.plot_options = ["performance", "kappa"]
elif self.task_type == "regression":
self.plot_options = ["mean_square_error", "true_vs_predict"]
elif self.task_type == "multi_output":
self.plot_options = ["hamming_score", "exact_match", "j_index"]
elif plot_options is not None:
self.plot_options = [x.lower() for x in plot_options]
for i in range(len(self.plot_options)):
if self.plot_options[i] not in PLOT_TYPES:
raise ValueError(str(self.plot_options[i]) + ": Plot type not supported.")
After Change
self.show_plot = show_plot
self.plot_options = None
if plot_options is None:
self.plot_options = TASK_PLOT_OPTIONS[self.task_type]
else:
self.plot_options = map(lower, plot_options)
for plot_option in self.plot_options:
if plot_option not in PLOT_TYPES: