diff --git a/docs/Makefile b/docs/Makefile
index 2b4de2d8..b9f1cf95 100644
--- a/docs/Makefile
+++ b/docs/Makefile
@@ -14,7 +14,7 @@ help:
 	@$(SPHINXBUILD) -M help "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)
 
 apidoc:
-	$(SPHINXAPIDOC) -efM -o source ../$(SPHINXPROJ)
+	$(SPHINXAPIDOC) -efM -o source ../$(SPHINXPROJ) && python3 format.py
 
 server:
 	cd build/html && python -m http.server
diff --git a/docs/format.py b/docs/format.py
new file mode 100644
index 00000000..7cc341c2
--- /dev/null
+++ b/docs/format.py
@@ -0,0 +1,65 @@
+import os
+
+
+def shorten(file, to_delete, cut=False):
+    if file.endswith("index.rst") or file.endswith("conf.py"):
+        return
+    res = []
+    with open(file, "r") as fin:
+        lines = fin.readlines()
+    for line in lines:
+        if cut and line.rstrip() == "Submodules":
+            break
+        else:
+            res.append(line.rstrip())
+    for i, line in enumerate(res):
+        if line.endswith(" package"):
+            res[i] = res[i][:-len(" package")]
+            res[i + 1] = res[i + 1][:-len(" package")]
+        elif line.endswith(" module"):
+            res[i] = res[i][:-len(" module")]
+            res[i + 1] = res[i + 1][:-len(" module")]
+        else:
+            for name in to_delete:
+                if line.endswith(name):
+                    res[i] = "del"
+
+    with open(file, "w") as fout:
+        for line in res:
+            if line != "del":
+                print(line, file=fout)
+
+
+def clear(path='./source/'):
+    files = os.listdir(path)
+    to_delete = [
+        "fastNLP.core.dist_trainer",
+        "fastNLP.core.predictor",
+
+        "fastNLP.io.file_reader",
+        "fastNLP.io.config_io",
+
+        "fastNLP.embeddings.contextual_embedding",
+
+        "fastNLP.modules.dropout",
+        "fastNLP.models.base_model",
+        "fastNLP.models.bert",
+        "fastNLP.models.enas_utils",
+        "fastNLP.models.enas_controller",
+        "fastNLP.models.enas_model",
+        "fastNLP.models.enas_trainer",
+    ]
+    for file in files:
+        if not os.path.isdir(path + file):
+            res = file.split('.')
+            if len(res) > 4:
+                to_delete.append(file[:-4])
+            elif len(res) == 4:
+                shorten(path + file, to_delete, True)
+            else:
+                shorten(path + file, to_delete)
+    for file in to_delete:
+        os.remove(path + file + ".rst")
+
+
+clear()