ANL-CEEESA/LLEPE
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/LLEPE.git synced 2025-12-06 01:48:53 -06:00
Code Issues Packages Projects Releases Wiki Activity

fixed parity plot bug for unlabeled, uncolored data

Browse Source
This commit is contained in:
titusquah
2020-07-14 12:48:41 -06:00
parent 1facd9bb42
commit ac8067baf4
2 changed files with 49 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

81
docs/Examples/1_getting_started.ipynb
View File

File diff suppressed because one or more lines are too long

10
llepe/llepe.py
View File

@@ -1267,6 +1267,7 @@ class LLEPE:
default_title = '{0} complex eq. conc. (mol/L)'.format( default_title = '{0} complex eq. conc. (mol/L)'.format(
compared_species) compared_species)
fig, ax = plt.subplots() fig, ax = plt.subplots()
if isinstance(data_labels, list): if isinstance(data_labels, list):
unique_labels = list(set(data_labels)) unique_labels = list(set(data_labels))
for label in unique_labels: for label in unique_labels:
@@ -1287,17 +1288,14 @@ class LLEPE:
if c_label is not None: if c_label is not None:
c_bar.set_label(c_label, rotation=270, labelpad=20) c_bar.set_label(c_label, rotation=270, labelpad=20)
else: else:
ax.scatter(meas, pred, color="r", ax.scatter(meas, pred, c="r", label="")
legend=False)
ax.plot(min_max_data, min_max_data, color="r", label="") ax.plot(min_max_data, min_max_data, color="b", label="")
if print_r_squared: if print_r_squared:
ax.text(min_max_data[0], ax.text(min_max_data[0],
min_max_data[1] * 0.9, min_max_data[1] * 0.9,
'$R^2$={0:.2f}'.format(self.r_squared(compared_value))) '$R^2$={0:.2f}'.format(self.r_squared(compared_value)))
# plt.legend(loc='lower right')
# else:
# plt.legend()
ax.set(xlabel='Measured', ylabel='Predicted') ax.set(xlabel='Measured', ylabel='Predicted')
if plot_title is None: if plot_title is None: