From 976563526a9c5e0dc6c63c5dc8f5a4edcbf5920d Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 21 Mar 2019 17:14:44 +0100 Subject: [PATCH] Added train and predict functions to module documentation Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 6f2990b..966177b 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -29,6 +29,10 @@ Classes: ``DropoutSSD``: wraps Dropout SSD 300 model ``SSD``: wraps vanilla SSD 300 model + +Functions: + predict(...): runs trained SSD/DropoutSSD on a given data set + train(...): trains the SSD/DropoutSSD on a given data set """ import os import pickle