修复::fix tang7 demo notebook

libwaifu · Aug 28, 2018 · 4361531 · 4361531
1 parent ea77f83
commit 4361531
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 7 deletions.
diff --git a/Models/WaifuSY-Tang.nb b/Models/WaifuSY-Tang.nb
diff --git a/Training/Tang-RNN/Tang.m b/Training/Tang-RNN/Tang.m
@@ -22,7 +22,9 @@
 (*这里应该填这个函数的介绍*)
 (* ::Section:: *)
 (*函数说明*)
-ExampleFunction::usage = "这里应该填这个函数的说明,如果要换行用\"\\r\"\r就像这样";
+Tang7::usage = "";
+Tang7Line::usage = "";
+Tang7LineRandom::usage = "";
 (* ::Section:: *)
 (*程序包正体*)
 (* ::Subsection::Closed:: *)
@@ -47,7 +49,7 @@
 Tang7LineRandom[char_String] := Block[
 	{state, next, head, top},
 	If[StringLength@char >= 8, Return[]];
-	state = NetStateObject[net];
+	state = NetStateObject[WaifuTang7];
 	next[str_, r_ : 50] := (
 		top = Association @@ state[str, {"TopProbabilities", r}];
 		top = KeyDrop[top, Characters["，。？兮" <> StringTake[str, -1]]];
@@ -56,7 +58,7 @@
 	head = Nest[next[#, 5$TangShiComplex]&, char, 7 - StringLength@char];
 	Nest[next[#, $TangShiComplex]&, head <> "，", 7] <> "。"
 ];
-Options[Tang7] = {Line -> 8};
+Options[Tang7] = {Line -> 8, Riffle -> True};
 Tang7[char_, OptionsPattern[]] := Block[
 	{next, shi, line},
 	line = OptionValue[Line];

diff --git a/Training/Tang-RNN/Tang.nb b/Training/Tang-RNN/Tang.nb