update ood code docs

mindarmour/reliability/__init__.py

@@ -12,9 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """
-Reliability methods of MindArmour
+Reliability methods of MindArmour.
 """
 
 from .model_fault_injection.fault_injection import FaultInjector
+from .concept_drift.concept_drift_check_time_series import ConceptDriftCheckTimeSeries
+from .concept_drift.concept_drift_check_images import OodDetector
+from .concept_drift.concept_drift_check_images import OodDetectorFeatureCluster
 
-__all__ = ['FaultInjector']
+__all__ = ['FaultInjector',
+           'ConceptDriftCheckTimeSeries',
+           'OodDetector',
+           'OodDetectorFeatureCluster']

mindarmour/reliability/concept_drift/concept_drift_check_images.py

@@ -72,6 +72,7 @@ class OodDetector:
     def ood_predict(self, threshold, ds_test):
         """
         The out-of-distribution detection.
+
         Args:
             threshold (float): the threshold to judge ood data. One can set value by experience
                 or use function get_optimal_threshold.
@@ -178,6 +179,7 @@ class OodDetectorFeatureCluster(OodDetector):
     def ood_predict(self, threshold, ds_test):
         """
         The out-of-distribution detection.
+
         Args:
             threshold (float): the threshold to judge ood data. One can set value by experience
                 or use function get_optimal_threshold.