Add some documentation

an1lam · an1lam · commit 7f21ea21e4b5 · 2022-06-18T09:00:11.000-04:00
diff --git a/src/tf_coop_in_silico_mutagenesis.py b/src/tf_coop_in_silico_mutagenesis.py
@@ -7,19 +7,14 @@
 import numpy as np
 import pandas as pd
 import seaborn as sns
-import simdna
 import statsmodels.api as sm
 import torch
 import uncertainty_toolbox.data as udata
 import uncertainty_toolbox.metrics as umetrics
 import uncertainty_toolbox.viz as uviz
-from IPython.display import clear_output
 from matplotlib import pyplot as plt
 from scipy import stats
-from simdna import synthetic
 from tqdm.auto import tqdm, trange
-from uncertainty_toolbox.metrics_calibration import \
-    get_proportion_lists_vectorized
 from uncertainty_toolbox.recalibration import iso_recal
 
 from ensemble import CalibratedRegressionEnsemble, Ensemble
diff --git a/src/tf_coop_main.py b/src/tf_coop_main.py
@@ -1,3 +1,7 @@
+"""
+Main file for running in-silico mutagenesis experiments using simulated data.
+"""
+
 import argparse
 import copy
 import json
