[UPDATE] update plots

hollydinkel · Aug 19, 2024 · 413821f · 413821f
1 parent 6d57b8c
commit 413821f
Show file tree

Hide file tree

Showing 20 changed files with 92 additions and 78 deletions.
diff --git a/images/asset_turnover.png b/images/asset_turnover.png
diff --git a/images/ebitda.png b/images/ebitda.png
diff --git a/images/ebitda_separate.png b/images/ebitda_separate.png
diff --git a/images/financial_leverage.png b/images/financial_leverage.png
diff --git a/images/fraction_constellation.png b/images/fraction_constellation.png
diff --git a/images/gom_isp.png b/images/gom_isp.png
diff --git a/images/iqps_isp.png b/images/iqps_isp.png
diff --git a/images/kleos_isp.png b/images/kleos_isp.png
diff --git a/images/npm.png b/images/npm.png
diff --git a/images/planet_isp.png b/images/planet_isp.png
diff --git a/images/revenue.png b/images/revenue.png
diff --git a/images/revenue_separate.png b/images/revenue_separate.png
diff --git a/images/roe.png b/images/roe.png
diff --git a/images/satellogic_isp.png b/images/satellogic_isp.png
diff --git a/images/total_asset_growth_rate.png b/images/total_asset_growth_rate.png
diff --git a/images/total_equity_growth_rate.png b/images/total_equity_growth_rate.png
diff --git a/images/total_liabilities_growth_rate.png b/images/total_liabilities_growth_rate.png
diff --git a/images/total_satellites.png b/images/total_satellites.png
diff --git a/src/plot_utils.py b/src/plot_utils.py
@@ -25,27 +25,30 @@ def plotKPIs(fig, ax, data, key, companyMetadata, company, ylabel):
     ax.plot(data["Quarter"], data[key], color=company_color, linestyle=companyMetadata[company]["style"], label=company, linewidth=10)
     ax.legend(ncol=3, bbox_to_anchor=(0.9, -0.15), fancybox=True, title="Company", fontsize='16', frameon=True)
     ax.set_xlim([pd.to_datetime('2018-06-01'), pd.to_datetime('2024-12-31')])
+    if ylabel=="Revenue ($USD)":
+        ax.set_yscale('log')
+        ax.set_ylim([1, 10**9])
     fig.tight_layout()
     box = ax.get_position()
     ax.set_position([box.x0, box.y0 + box.height * 0.2, box.width, box.height * 0.8])
     plt.rcParams['legend.title_fontsize'] = '16'
     return fig
 
-def plotPredictions(fig, ax, company, companyMetadata, time=None, inSamplePrediction=None, outSamplePrediction=None, ground_truth=None, split=None, plotEverything=False):
+def plotPredictions(fig, ax, company, companyMetadata, time=None, inSamplePrediction=None, outSamplePrediction=None, ground_truth=None, ylabel=None, split=None, plotEverything=False):
     size = 22
     if company != "Synspective" and not plotEverything:
         company_color = np.array(companyMetadata[company]["color"]) / 255
         ax.set_title(company, fontsize=16, fontweight='bold')
         ax.tick_params(axis='both', which='major', labelsize=16)
         ax.set_xlabel('Year', fontsize=16)
-        ax.set_ylabel('Revenue ($ USD)', fontsize=16)
+        ax.set_ylabel(ylabel, fontsize=16)
         ax.xaxis.set_major_locator(ticker.MaxNLocator(nbins=5))
         ax.yaxis.set_major_locator(ticker.MaxNLocator(nbins=5))
         ax.set_xlim([pd.to_datetime('2018-06-01'), pd.to_datetime('2024-12-31')])
         ax.plot(time[:-split], inSamplePrediction, color=company_color, linestyle="dotted", label="In-Sample Prediction", linewidth=10)
         ax.plot(time[-split:], outSamplePrediction, color=company_color, alpha=0.5, linestyle="dashed", label="Out-of-Sample Prediction", linewidth=10)
         ax.plot(time, ground_truth, color=company_color, linestyle="solid", label="Ground Truth", linewidth=10)
-        ax.legend(ncol=2, bbox_to_anchor=(0.85, -0.15), fancybox=True, fontsize='16', frameon=True)
+        ax.legend(ncol=3, bbox_to_anchor=(1.0, -0.15), fancybox=True, fontsize='14', frameon=True)
         box = ax.get_position()
         ax.set_position([box.x0, box.y0 + box.height * 0.2, box.width, box.height * 0.8])
     elif company!= "Synspective" and plotEverything:
@@ -77,6 +80,6 @@ def plotPredictions(fig, ax, company, companyMetadata, time=None, inSamplePredic
         line2, = ax.plot([], [], color='k', linestyle="dashed", alpha=0.5, linewidth=12, label='Out-of-Sample Prediction')
         line3, = ax.plot([], [], color='k', linestyle="solid", linewidth=12, label='Ground Truth Data')
         # Add the legend with specified labels
-        fig.legend([line1, line2, line3], ['In-Sample Prediction', 'Out-of-Sample Prediction', 'Ground Truth Data'],
-           loc='upper center', bbox_to_anchor=(0.5, 0.08), ncol=2, prop={'size': size})
+        fig.legend([line1, line2, line3], ['In-Sam1le Prediction', 'Out-of-Sample Prediction', 'Ground Truth Data'],
+           loc='upper center', bbox_to_anchor=(0.5, 0.08), ncol=3, prop={'size': size})
     return fig
diff --git a/src/process.py b/src/process.py