diff --git a/pykt/assist09/coldstart/pykt-qdkt-csNall-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-qdkt-csNall-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a2c5dcadc11d6c496a27f9c2814aa6e8145c4930
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qdkt-csNall-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 305.3, "epochs": 54, "best_epoch": 44, "dev_auc_pykt": 0.6794621742777567, "n_train_users": 1730, "n_train_chunks": 2070, "n_dev_chunks": 501, "batch_size": 32, "seq_len": 200, "n_val_users": 432, "score_s": 15.4, "flip_max_diff": 0.0, "peak_gb": 3.86}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qdkt-csNall-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-qdkt-csNall-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a279b0096cfb29940e90eb8c6d17bccbdccbca55
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qdkt-csNall-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 191.9, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.67701440011298, "n_train_users": 1730, "n_train_chunks": 2060, "n_dev_chunks": 511, "batch_size": 32, "seq_len": 200, "n_val_users": 432, "score_s": 18.8, "flip_max_diff": 0.0, "peak_gb": 3.86}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qdkt-csNall-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-qdkt-csNall-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..fbaa766bed46063126f39b245a28afb27ca5777a
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qdkt-csNall-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 170.5, "epochs": 52, "best_epoch": 42, "dev_auc_pykt": 0.682065901343358, "n_train_users": 1730, "n_train_chunks": 2051, "n_dev_chunks": 520, "batch_size": 32, "seq_len": 200, "n_val_users": 432, "score_s": 11.3, "flip_max_diff": 0.0, "peak_gb": 3.86}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qdkt-csNall-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-qdkt-csNall-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..587ad50e5bec7e57a80143fb4f0ef8edb799c55a
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qdkt-csNall-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 154.1, "epochs": 50, "best_epoch": 40, "dev_auc_pykt": 0.6833651767828983, "n_train_users": 1730, "n_train_chunks": 2062, "n_dev_chunks": 509, "batch_size": 32, "seq_len": 200, "n_val_users": 432, "score_s": 11.4, "flip_max_diff": 0.0, "peak_gb": 3.86}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qdkt-csNall-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-qdkt-csNall-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..215410a86fa1a23660785732c9f8d29ef1832f54
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qdkt-csNall-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 149.4, "epochs": 47, "best_epoch": 37, "dev_auc_pykt": 0.685057258220471, "n_train_users": 1730, "n_train_chunks": 2039, "n_dev_chunks": 532, "batch_size": 32, "seq_len": 200, "n_val_users": 432, "score_s": 11.8, "flip_max_diff": 0.0, "peak_gb": 3.86}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qdkt-ics0/train_info.json.done b/pykt/assist09/coldstart/pykt-qdkt-ics0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..26f524e9812aa7475d0146ee29fbca1bd09a3619
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qdkt-ics0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 323.4, "epochs": 47, "best_epoch": 37, "dev_auc_pykt": 0.672166416331359, "n_train_users": 1730, "n_train_chunks": 1967, "n_dev_chunks": 481, "batch_size": 16, "seq_len": 200, "n_val_users": 432, "score_s": 1.3, "flip_max_diff": 0.0, "peak_gb": 3.72}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qdkt_bsprobe-csN32-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-qdkt_bsprobe-csN32-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..87d2d7bc521834436d42c452961f23252cf1e23a
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qdkt_bsprobe-csN32-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.7, "epochs": 33, "best_epoch": 23, "dev_auc_pykt": 0.5145699206088414, "n_train_users": 26, "n_train_chunks": 31, "n_dev_chunks": 9, "batch_size": 32, "seq_len": 200, "n_val_users": 6, "score_s": 9.6, "flip_max_diff": 0.0, "peak_gb": 3.86}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN128-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN128-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..fe9d7e65896d5ff1abf76c843decf7c9b59f746b
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN128-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.1, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.6654778552345968, "n_train_users": 102, "n_train_chunks": 115, "n_dev_chunks": 29, "batch_size": 32, "seq_len": 200, "n_val_users": 26, "score_s": 19.3, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN128-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN128-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..d44b0459ce8bb49059acc351ab1c07f08e7ccb91
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN128-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.3, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.6478421052631579, "n_train_users": 102, "n_train_chunks": 120, "n_dev_chunks": 28, "batch_size": 32, "seq_len": 200, "n_val_users": 26, "score_s": 28.6, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN128-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN128-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..646d2592673027120d53371f2221078999ffcb6b
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN128-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 10.2, "epochs": 24, "best_epoch": 14, "dev_auc_pykt": 0.667894646843832, "n_train_users": 102, "n_train_chunks": 111, "n_dev_chunks": 37, "batch_size": 32, "seq_len": 200, "n_val_users": 26, "score_s": 18.2, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN128-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN128-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..797f0a545058fe7eb86e316af14698b3ee0e2d4f
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN128-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.4, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.6745577523413111, "n_train_users": 102, "n_train_chunks": 124, "n_dev_chunks": 32, "batch_size": 32, "seq_len": 200, "n_val_users": 26, "score_s": 15.9, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN128-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN128-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1472066458016639bd52da00cd5fdf7f86ef6db5
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN128-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.0, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.673960216281356, "n_train_users": 102, "n_train_chunks": 115, "n_dev_chunks": 30, "batch_size": 32, "seq_len": 200, "n_val_users": 26, "score_s": 15.5, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN16-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN16-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8c3d7b7cc262f23f6c84abe4b812a9611d833597
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN16-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 14.2, "epochs": 27, "best_epoch": 17, "dev_auc_pykt": 0.6373735302160239, "n_train_users": 13, "n_train_chunks": 17, "n_dev_chunks": 4, "batch_size": 32, "seq_len": 200, "n_val_users": 3, "score_s": 15.5, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN16-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN16-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..fae7d4690f5b44f652748cabef331e9d36093cd3
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN16-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 8.2, "epochs": 34, "best_epoch": 24, "dev_auc_pykt": 0.49224328983009114, "n_train_users": 13, "n_train_chunks": 18, "n_dev_chunks": 3, "batch_size": 32, "seq_len": 200, "n_val_users": 3, "score_s": 22.0, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN16-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN16-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..123921df63df9ae6cce1dfa3d1e2eb356d35a692
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN16-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.3, "epochs": 17, "best_epoch": 7, "dev_auc_pykt": 0.5931293612453032, "n_train_users": 13, "n_train_chunks": 13, "n_dev_chunks": 3, "batch_size": 32, "seq_len": 200, "n_val_users": 3, "score_s": 14.7, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN16-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN16-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a746472d2d5911fd95c810e961d914641094ddfb
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN16-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.6, "epochs": 19, "best_epoch": 9, "dev_auc_pykt": 0.4881794078656651, "n_train_users": 13, "n_train_chunks": 15, "n_dev_chunks": 3, "batch_size": 32, "seq_len": 200, "n_val_users": 3, "score_s": 15.7, "flip_max_diff": 0.0, "peak_gb": 4.11}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN16-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN16-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..d0361e08b4b26ecb3b8b9c6f1b7654b06014a22b
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN16-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4.6, "epochs": 23, "best_epoch": 13, "dev_auc_pykt": 0.6033525954785797, "n_train_users": 13, "n_train_chunks": 15, "n_dev_chunks": 5, "batch_size": 32, "seq_len": 200, "n_val_users": 3, "score_s": 14.6, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN256-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN256-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..94e16661e49b680bdbae1f2503a8ecd2e218d556
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN256-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 20.9, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.7040927947887065, "n_train_users": 205, "n_train_chunks": 235, "n_dev_chunks": 64, "batch_size": 32, "seq_len": 200, "n_val_users": 51, "score_s": 16.6, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN256-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN256-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..48b7c5ace73a0884e94ff5e6514c0474c5625b2f
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN256-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 13.3, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.6506889882957485, "n_train_users": 205, "n_train_chunks": 237, "n_dev_chunks": 59, "batch_size": 32, "seq_len": 200, "n_val_users": 51, "score_s": 21.6, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN256-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN256-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ba4f77ae131bc53cb46f16dc9c64842d1e210011
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN256-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 16.7, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.7365543334576146, "n_train_users": 205, "n_train_chunks": 244, "n_dev_chunks": 59, "batch_size": 32, "seq_len": 200, "n_val_users": 51, "score_s": 20.8, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN256-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN256-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..44eb51ed09bc62886ccd94d588a835c4263c9e20
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN256-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 38.5, "epochs": 27, "best_epoch": 17, "dev_auc_pykt": 0.6767781935755315, "n_train_users": 205, "n_train_chunks": 249, "n_dev_chunks": 59, "batch_size": 32, "seq_len": 200, "n_val_users": 51, "score_s": 24.5, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN256-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN256-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..686ec042d4e0e39091ce918f4fa92913abfa82dc
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN256-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 28.9, "epochs": 20, "best_epoch": 10, "dev_auc_pykt": 0.712794827253426, "n_train_users": 205, "n_train_chunks": 234, "n_dev_chunks": 60, "batch_size": 32, "seq_len": 200, "n_val_users": 51, "score_s": 26.8, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN32-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN32-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..80a0b9b69b08ebd379500e86c89383eff1bc1910
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN32-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4.2, "epochs": 19, "best_epoch": 9, "dev_auc_pykt": 0.6421228118921921, "n_train_users": 26, "n_train_chunks": 31, "n_dev_chunks": 6, "batch_size": 32, "seq_len": 200, "n_val_users": 6, "score_s": 15.2, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN32-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN32-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b5e75691db31f3ee0de8cb74549602b0676a1292
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN32-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 5.2, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.6547194534707771, "n_train_users": 26, "n_train_chunks": 31, "n_dev_chunks": 9, "batch_size": 8, "seq_len": 200, "n_val_users": 6, "score_s": 25.8, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN32-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN32-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..0d33fafbb461c929d599edb17ab45154eda0aefd
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN32-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 5.5, "epochs": 29, "best_epoch": 19, "dev_auc_pykt": 0.6311799604482531, "n_train_users": 26, "n_train_chunks": 27, "n_dev_chunks": 6, "batch_size": 32, "seq_len": 200, "n_val_users": 6, "score_s": 21.0, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN32-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN32-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2baa41887bba39e7fd08094eacf399dc8ca92172
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN32-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3.2, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.6434898880160453, "n_train_users": 26, "n_train_chunks": 30, "n_dev_chunks": 6, "batch_size": 32, "seq_len": 200, "n_val_users": 6, "score_s": 28.8, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN32-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN32-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..bc10a3cbddad5ee36291369afd6673bfe3b46ad6
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN32-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4.4, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.5464880952380953, "n_train_users": 26, "n_train_chunks": 30, "n_dev_chunks": 6, "batch_size": 32, "seq_len": 200, "n_val_users": 6, "score_s": 15.2, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN64-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN64-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..c5ce3784670f98dd5756a3796ffa4076864da438
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN64-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 10.9, "epochs": 40, "best_epoch": 30, "dev_auc_pykt": 0.6012012838147639, "n_train_users": 51, "n_train_chunks": 57, "n_dev_chunks": 15, "batch_size": 32, "seq_len": 200, "n_val_users": 13, "score_s": 15.4, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN64-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN64-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..d7b6815a9666f548be00411f34c4ee0feb573a11
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN64-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 7.2, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.590452101844247, "n_train_users": 51, "n_train_chunks": 62, "n_dev_chunks": 18, "batch_size": 32, "seq_len": 200, "n_val_users": 13, "score_s": 20.9, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN64-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN64-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a63b077c2799089553e594dc6fa1b899c19fcad3
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN64-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 5.1, "epochs": 17, "best_epoch": 7, "dev_auc_pykt": 0.6405007206533924, "n_train_users": 51, "n_train_chunks": 56, "n_dev_chunks": 14, "batch_size": 32, "seq_len": 200, "n_val_users": 13, "score_s": 21.5, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN64-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN64-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..84069195d104a06ce704aa2422caec5775483067
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN64-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 15.8, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.6089209151504233, "n_train_users": 51, "n_train_chunks": 56, "n_dev_chunks": 16, "batch_size": 32, "seq_len": 200, "n_val_users": 13, "score_s": 16.0, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN64-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN64-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..c02f083d1c0db368552b4b1dcc10ec81cc5a9134
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN64-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4.3, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.6324653538166582, "n_train_users": 51, "n_train_chunks": 57, "n_dev_chunks": 16, "batch_size": 32, "seq_len": 200, "n_val_users": 13, "score_s": 15.4, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN8-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN8-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..79603505498185ddc116f76295f33282dddd7204
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN8-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.4, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.6090909090909091, "n_train_users": 6, "n_train_chunks": 10, "n_dev_chunks": 2, "batch_size": 32, "seq_len": 200, "n_val_users": 2, "score_s": 16.7, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN8-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN8-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..598807ae9020ac170ee7e17d5674c51d7a3c3fa5
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN8-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3.5, "epochs": 21, "best_epoch": 11, "dev_auc_pykt": 0.6314612033439303, "n_train_users": 6, "n_train_chunks": 8, "n_dev_chunks": 2, "batch_size": 32, "seq_len": 200, "n_val_users": 2, "score_s": 22.0, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN8-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN8-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..9980610fccee5636a145f6b9220c6a0d447f82d9
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN8-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.7, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.4594594594594595, "n_train_users": 6, "n_train_chunks": 6, "n_dev_chunks": 2, "batch_size": 32, "seq_len": 200, "n_val_users": 2, "score_s": 13.8, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN8-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN8-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..09170fb324ac6abf99aa90af314151011ae30d7d
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN8-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.7, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6875, "n_train_users": 6, "n_train_chunks": 6, "n_dev_chunks": 2, "batch_size": 32, "seq_len": 200, "n_val_users": 2, "score_s": 15.5, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csN8-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csN8-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e8f1c21d81a0698e1892917ecbd6a0e3159325fc
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csN8-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.5, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6051136363636364, "n_train_users": 6, "n_train_chunks": 6, "n_dev_chunks": 2, "batch_size": 32, "seq_len": 200, "n_val_users": 2, "score_s": 24.8, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csNall-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csNall-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..adc7e99ccef4d6f7aa1a91a1a120b886e9f8f778
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csNall-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 208.7, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.7630964722764355, "n_train_users": 1730, "n_train_chunks": 2070, "n_dev_chunks": 501, "batch_size": 32, "seq_len": 200, "n_val_users": 432, "score_s": 30.5, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csNall-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csNall-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e62c6ce7ddc02bd86c3dcf59859d5ccfbfdb7896
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csNall-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 213.4, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.7492894802266379, "n_train_users": 1730, "n_train_chunks": 2060, "n_dev_chunks": 511, "batch_size": 32, "seq_len": 200, "n_val_users": 432, "score_s": 29.8, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csNall-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csNall-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..59eda5ad0bd8edebd66d2c73e33ce9498506cd93
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csNall-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 176.6, "epochs": 16, "best_epoch": 6, "dev_auc_pykt": 0.7592538490732132, "n_train_users": 1730, "n_train_chunks": 2051, "n_dev_chunks": 520, "batch_size": 32, "seq_len": 200, "n_val_users": 432, "score_s": 26.7, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csNall-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csNall-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..63cb0254a53d49fe3e62bc15cf299d6f463b6741
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csNall-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 89.2, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.7589899694896364, "n_train_users": 1730, "n_train_chunks": 2062, "n_dev_chunks": 509, "batch_size": 32, "seq_len": 200, "n_val_users": 432, "score_s": 15.4, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-csNall-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-csNall-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..dcbc67d5e2f517e17b2f3bade033d5dfc0187432
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-csNall-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 122.7, "epochs": 19, "best_epoch": 9, "dev_auc_pykt": 0.7514122069958485, "n_train_users": 1730, "n_train_chunks": 2039, "n_dev_chunks": 532, "batch_size": 32, "seq_len": 200, "n_val_users": 432, "score_s": 16.7, "flip_max_diff": 0.0, "peak_gb": 4.03}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-qikt-ics0/train_info.json.done b/pykt/assist09/coldstart/pykt-qikt-ics0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..9d0c2c0309096eafd4e87538b0dee70d81ae9fe8
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-qikt-ics0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 217.2, "epochs": 21, "best_epoch": 11, "dev_auc_pykt": 0.7580522510831472, "n_train_users": 1730, "n_train_chunks": 1967, "n_dev_chunks": 481, "batch_size": 32, "seq_len": 200, "n_val_users": 432, "score_s": 1.4, "flip_max_diff": 0.0, "peak_gb": 3.89}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN128-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN128-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b3f789f569259d39656fc63c24d6423732f8d956
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN128-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 68.7, "epochs": 29, "best_epoch": 19, "dev_auc_pykt": 0.6495709394869058, "n_train_users": 102, "n_train_chunks": 122, "n_dev_chunks": 30, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 57.6, "flip_max_diff": 0.0, "peak_gb": 0.59}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN128-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN128-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..66264c91fac7588e13da69cdd57ff8db558e86ce
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN128-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 46.7, "epochs": 27, "best_epoch": 17, "dev_auc_pykt": 0.6907275774114117, "n_train_users": 102, "n_train_chunks": 119, "n_dev_chunks": 31, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 35.1, "flip_max_diff": 0.0, "peak_gb": 0.58}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN16-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN16-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..4056f13103809e32bd7326df2e7fc9441c69cfa8
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN16-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 18.9, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6041129360662413, "n_train_users": 13, "n_train_chunks": 19, "n_dev_chunks": 4, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 36.4, "flip_max_diff": 0.0, "peak_gb": 0.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN16-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN16-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..f73c6a38b61a68bab3bdfb83deff5fcbd7d951b6
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN16-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 25.5, "epochs": 17, "best_epoch": 7, "dev_auc_pykt": 0.6081686429512517, "n_train_users": 13, "n_train_chunks": 13, "n_dev_chunks": 3, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 29.6, "flip_max_diff": 0.0, "peak_gb": 0.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN16-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN16-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8af76c61241e2d21a990abd479cc389bc3b28301
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN16-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 33.6, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.5763346354166666, "n_train_users": 13, "n_train_chunks": 15, "n_dev_chunks": 6, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 54.6, "flip_max_diff": 0.0, "peak_gb": 0.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN256-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN256-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6f6dcaf83432ba62df8346c8ef6ca27c342dc314
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN256-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 83.0, "epochs": 31, "best_epoch": 21, "dev_auc_pykt": 0.7164140691365205, "n_train_users": 205, "n_train_chunks": 246, "n_dev_chunks": 66, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 58.5, "flip_max_diff": 0.0, "peak_gb": 1.15}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN256-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN256-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..5d6b915ad2402d2fff7a3fd58329c85bff55815a
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN256-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 105.3, "epochs": 48, "best_epoch": 38, "dev_auc_pykt": 0.6947745041370403, "n_train_users": 205, "n_train_chunks": 245, "n_dev_chunks": 63, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 46.0, "flip_max_diff": 0.0, "peak_gb": 1.15}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN32-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN32-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7bdd68a6b8769689b189166aa87f1cf32b401b81
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN32-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 38.5, "epochs": 17, "best_epoch": 7, "dev_auc_pykt": 0.5971103084190054, "n_train_users": 26, "n_train_chunks": 33, "n_dev_chunks": 6, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 55.8, "flip_max_diff": 0.0, "peak_gb": 0.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN32-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN32-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3828dd72efe91e11c98f75086ccf9f9f567c16aa
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN32-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 21.3, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.7508023106546854, "n_train_users": 26, "n_train_chunks": 28, "n_dev_chunks": 6, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 34.2, "flip_max_diff": 0.0, "peak_gb": 0.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN32-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN32-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..966cf29b838b371518a1893ba377dceee8df38d9
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN32-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 89.5, "epochs": 53, "best_epoch": 43, "dev_auc_pykt": 0.6417989417989418, "n_train_users": 26, "n_train_chunks": 31, "n_dev_chunks": 6, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 29.8, "flip_max_diff": 0.0, "peak_gb": 0.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN64-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN64-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..9a6915eb917c5b77a92b14af7f47fbbb1fde3653
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN64-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 30.2, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.5522161172161172, "n_train_users": 51, "n_train_chunks": 60, "n_dev_chunks": 16, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 57.9, "flip_max_diff": 0.0, "peak_gb": 0.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN64-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN64-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6d75d36d0da972484ef9cc0a6ac8282240a85a95
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN64-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 44.4, "epochs": 27, "best_epoch": 17, "dev_auc_pykt": 0.6716386945766949, "n_train_users": 51, "n_train_chunks": 59, "n_dev_chunks": 15, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 34.7, "flip_max_diff": 0.0, "peak_gb": 0.33}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN64-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN64-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..86f39d6d4050bd501d5ac496f084e74b2c6fbda2
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN64-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 71.7, "epochs": 46, "best_epoch": 36, "dev_auc_pykt": 0.6802416052416054, "n_train_users": 51, "n_train_chunks": 58, "n_dev_chunks": 16, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 29.8, "flip_max_diff": 0.0, "peak_gb": 0.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN8-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN8-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2e3560fecae1bcb2c3f46c5b46595d3af4a5b3ff
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN8-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 34.7, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.6625668449197861, "n_train_users": 6, "n_train_chunks": 11, "n_dev_chunks": 2, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 60.0, "flip_max_diff": 0.0, "peak_gb": 0.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN8-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN8-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..87416ab666d5e24f833bc14735fe7edeebf12d0f
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN8-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 18.5, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.5481691368788144, "n_train_users": 6, "n_train_chunks": 6, "n_dev_chunks": 2, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 31.3, "flip_max_diff": 0.0, "peak_gb": 0.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csN8-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csN8-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ce9e3c98d3d928ab970e9e570ef44cd6eaccf5bd
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csN8-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 20.1, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.6278409090909092, "n_train_users": 6, "n_train_chunks": 6, "n_dev_chunks": 2, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 39.1, "flip_max_diff": 0.0, "peak_gb": 0.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csNall-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csNall-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..5de34edceb7368329f371f18594513d23ed31a77
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csNall-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 856.7, "epochs": 46, "best_epoch": 36, "dev_auc_pykt": 0.7714855511943882, "n_train_users": 1730, "n_train_chunks": 2191, "n_dev_chunks": 525, "batch_size": 256, "seq_len": 200, "n_val_users": 432, "score_s": 57.5, "flip_max_diff": 0.0, "peak_gb": 1.2}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-csNall-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-csNall-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..33a9efd02801f8e6b3868b61a53bce0f335412e7
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-csNall-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 822.9, "epochs": 51, "best_epoch": 41, "dev_auc_pykt": 0.7654764701904608, "n_train_users": 1730, "n_train_chunks": 2151, "n_dev_chunks": 565, "batch_size": 256, "seq_len": 200, "n_val_users": 432, "score_s": 44.1, "flip_max_diff": 0.0, "peak_gb": 1.2}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-ics1/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-ics1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a76a6fac5922e153d46336d81866fe178e39992b
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-ics1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 723.7, "epochs": 42, "best_epoch": 32, "dev_auc_pykt": 0.7486215702530099, "n_train_users": 1730, "n_train_chunks": 2051, "n_dev_chunks": 509, "batch_size": 256, "seq_len": 200, "n_val_users": 432, "score_s": 3.2, "flip_max_diff": 0.0, "peak_gb": 1.2}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-rekt-ics2/train_info.json.done b/pykt/assist09/coldstart/pykt-rekt-ics2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6adc11c196fbf4d20db02dad7d34b295a78e3464
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-rekt-ics2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 730.4, "epochs": 48, "best_epoch": 38, "dev_auc_pykt": 0.7647948880276028, "n_train_users": 1730, "n_train_chunks": 2039, "n_dev_chunks": 518, "batch_size": 256, "seq_len": 200, "n_val_users": 432, "score_s": 2.9, "flip_max_diff": 0.0, "peak_gb": 1.2}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN128-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN128-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..d7edcf9578b3eecad64cdd376fb763e33421dcaf
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN128-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 24.8, "epochs": 31, "best_epoch": 21, "dev_auc_pykt": 0.6902342133434571, "n_train_users": 102, "n_train_chunks": 122, "n_dev_chunks": 30, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 41.7, "flip_max_diff": 0.0, "peak_gb": 9.31}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN128-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN128-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..4776ae469f6fd3d8b484d0955e039f48186a203a
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN128-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 18.4, "epochs": 27, "best_epoch": 17, "dev_auc_pykt": 0.7078491683754842, "n_train_users": 102, "n_train_chunks": 127, "n_dev_chunks": 28, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 42.4, "flip_max_diff": 0.0, "peak_gb": 9.31}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN128-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN128-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..f8f6518d53d51cb46a46a478d8aaa088055359b0
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN128-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 29.0, "epochs": 41, "best_epoch": 31, "dev_auc_pykt": 0.7477509790996161, "n_train_users": 102, "n_train_chunks": 116, "n_dev_chunks": 39, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 31.2, "flip_max_diff": 0.0, "peak_gb": 9.31}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN128-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN128-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..10076b305e0b43d03d69e36f519d49f5bdda1d60
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN128-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 16.3, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.7463584188605455, "n_train_users": 102, "n_train_chunks": 130, "n_dev_chunks": 35, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 50.5, "flip_max_diff": 0.0, "peak_gb": 9.31}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN128-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN128-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..4419a17264afdbcacd233b52c6d687fc91ae4784
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN128-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 7.3, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.7451074978954437, "n_train_users": 102, "n_train_chunks": 119, "n_dev_chunks": 31, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 26.3, "flip_max_diff": 0.0, "peak_gb": 9.31}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN16-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN16-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..bb739442d676f0bbd3327efc9072509e5bf107cc
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN16-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 10.6, "epochs": 20, "best_epoch": 10, "dev_auc_pykt": 0.587925885706529, "n_train_users": 13, "n_train_chunks": 19, "n_dev_chunks": 4, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 75.1, "flip_max_diff": 0.0, "peak_gb": 2.83}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN16-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN16-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7b08f6f69f34f99aa355a902326f9d1c372da648
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN16-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 13.2, "epochs": 40, "best_epoch": 30, "dev_auc_pykt": 0.6213363139592648, "n_train_users": 13, "n_train_chunks": 18, "n_dev_chunks": 4, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 36.6, "flip_max_diff": 0.0, "peak_gb": 2.68}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN16-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN16-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..5ab469567189e156ca01d110be5d9badc464186a
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN16-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 15.6, "epochs": 50, "best_epoch": 40, "dev_auc_pykt": 0.6361001317523056, "n_train_users": 13, "n_train_chunks": 13, "n_dev_chunks": 3, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 24.9, "flip_max_diff": 0.0, "peak_gb": 1.96}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN16-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN16-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e80eb4b94f166417a0d1c66ca59287f5152f5d88
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN16-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 7.1, "epochs": 21, "best_epoch": 11, "dev_auc_pykt": 0.6872514361467079, "n_train_users": 13, "n_train_chunks": 16, "n_dev_chunks": 3, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 32.9, "flip_max_diff": 0.0, "peak_gb": 2.4}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN16-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN16-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a56214e8340a2b6235e17b612250836b302cb211
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN16-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 8.5, "epochs": 32, "best_epoch": 22, "dev_auc_pykt": 0.5713704427083333, "n_train_users": 13, "n_train_chunks": 15, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 42.0, "flip_max_diff": 0.0, "peak_gb": 2.25}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN256-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN256-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..c5942cec01551b6f08cc9206969312f67c0dc7a0
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN256-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 55.4, "epochs": 32, "best_epoch": 22, "dev_auc_pykt": 0.7347598894917693, "n_train_users": 205, "n_train_chunks": 246, "n_dev_chunks": 66, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 42.3, "flip_max_diff": 0.0, "peak_gb": 9.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN256-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN256-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ced5352634eded33cdfa5c203f3eb1dc4e99136e
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN256-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 25.9, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.6753436360490078, "n_train_users": 205, "n_train_chunks": 247, "n_dev_chunks": 64, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 40.3, "flip_max_diff": 0.0, "peak_gb": 9.31}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN256-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN256-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1345b5c485a8ae76e34a1b48b28bff135f641e73
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN256-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 29.0, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.7491900779804317, "n_train_users": 205, "n_train_chunks": 261, "n_dev_chunks": 60, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 45.1, "flip_max_diff": 0.0, "peak_gb": 9.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN256-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN256-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..09ed56b588c425bb8f8ded44f1d925f655346cb2
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN256-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 98.0, "epochs": 56, "best_epoch": 46, "dev_auc_pykt": 0.7028241309672576, "n_train_users": 205, "n_train_chunks": 265, "n_dev_chunks": 62, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 33.8, "flip_max_diff": 0.0, "peak_gb": 9.31}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN256-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN256-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..96cedfbe72d132f6a46502d63c32437f855b8cff
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN256-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 35.8, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.6953758488794934, "n_train_users": 205, "n_train_chunks": 245, "n_dev_chunks": 63, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 45.1, "flip_max_diff": 0.0, "peak_gb": 9.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN32-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN32-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ab913bc9d833b774ebacfcb299e2d150c56fad43
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN32-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.3, "epochs": 34, "best_epoch": 24, "dev_auc_pykt": 0.6375382050569603, "n_train_users": 26, "n_train_chunks": 33, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 42.3, "flip_max_diff": 0.0, "peak_gb": 0.65}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN32-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN32-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7db6cb9aea7e2cf964b2f5af2aa15b06d0fe460e
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN32-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 10.4, "epochs": 26, "best_epoch": 16, "dev_auc_pykt": 0.7200806399897397, "n_train_users": 26, "n_train_chunks": 32, "n_dev_chunks": 10, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 39.1, "flip_max_diff": 0.0, "peak_gb": 4.7}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN32-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN32-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b2c16fc3b8bd088d178b117700e8bcbc34c64270
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN32-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.8, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.8140243902439024, "n_train_users": 26, "n_train_chunks": 28, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 32.2, "flip_max_diff": 0.0, "peak_gb": 4.12}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN32-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN32-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..33403075c0f6947d61766fbd9b25f19a6a2e7515
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN32-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 8.8, "epochs": 28, "best_epoch": 18, "dev_auc_pykt": 0.595437071703159, "n_train_users": 26, "n_train_chunks": 32, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 39.0, "flip_max_diff": 0.0, "peak_gb": 4.7}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN32-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN32-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6cce4a435a9130a4fdf68695bd8ea923e5ed16af
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN32-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.0, "epochs": 19, "best_epoch": 9, "dev_auc_pykt": 0.6108171663727219, "n_train_users": 26, "n_train_chunks": 31, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 24.2, "flip_max_diff": 0.0, "peak_gb": 4.57}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN64-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN64-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ee6b922f0034c5f10865dce3cd8e37ea2d7baf80
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN64-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 11.3, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.6408882783882783, "n_train_users": 51, "n_train_chunks": 60, "n_dev_chunks": 16, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 40.2, "flip_max_diff": 0.0, "peak_gb": 8.73}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN64-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN64-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..00ddbe6bd00033ecdb1ce28a94f8766bb4a1fbb1
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN64-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 15.8, "epochs": 21, "best_epoch": 11, "dev_auc_pykt": 0.6635036754311563, "n_train_users": 51, "n_train_chunks": 65, "n_dev_chunks": 20, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 38.4, "flip_max_diff": 0.0, "peak_gb": 9.31}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN64-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN64-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..0740d5cfb5515f2da0b0ae532d5839f2898158ab
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN64-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 11.2, "epochs": 28, "best_epoch": 18, "dev_auc_pykt": 0.7082293800037237, "n_train_users": 51, "n_train_chunks": 59, "n_dev_chunks": 15, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 43.6, "flip_max_diff": 0.0, "peak_gb": 8.6}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN64-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN64-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..4b519aa979cfe149b825a491450aa1213cb83a70
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN64-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.6, "epochs": 19, "best_epoch": 9, "dev_auc_pykt": 0.6625300340684956, "n_train_users": 51, "n_train_chunks": 59, "n_dev_chunks": 16, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 41.4, "flip_max_diff": 0.0, "peak_gb": 8.61}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN64-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN64-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e15f1fe02c6095bf1dc0019c783da0d52498992c
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN64-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 31.2, "epochs": 20, "best_epoch": 10, "dev_auc_pykt": 0.701036036036036, "n_train_users": 51, "n_train_chunks": 58, "n_dev_chunks": 16, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 23.8, "flip_max_diff": 0.0, "peak_gb": 8.48}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN8-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN8-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..21e290083977fb70b1c8178ae270bd8ccf4eeab3
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN8-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 13.7, "epochs": 37, "best_epoch": 27, "dev_auc_pykt": 0.6176470588235294, "n_train_users": 6, "n_train_chunks": 11, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 65.7, "flip_max_diff": 0.0, "peak_gb": 1.68}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN8-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN8-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..68f63e3ff281b082b7d6ef04b0da611f04d7a065
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN8-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 5.1, "epochs": 20, "best_epoch": 10, "dev_auc_pykt": 0.6983625730994152, "n_train_users": 6, "n_train_chunks": 8, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 29.5, "flip_max_diff": 0.0, "peak_gb": 1.25}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN8-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN8-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..fad3581ebc8ff00388df91cf88591ddaec0ba514
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN8-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4.6, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.5368352223190933, "n_train_users": 6, "n_train_chunks": 6, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 24.4, "flip_max_diff": 0.0, "peak_gb": 0.96}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN8-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN8-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2319fcb4ff4a67a80c858e96389e04c29f393718
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN8-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 7.0, "epochs": 27, "best_epoch": 17, "dev_auc_pykt": 0.752, "n_train_users": 6, "n_train_chunks": 7, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 34.3, "flip_max_diff": 0.0, "peak_gb": 1.12}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csN8-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csN8-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e2a1c25ef6fa45c2092561967b0875996294db80
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csN8-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4.3, "epochs": 23, "best_epoch": 13, "dev_auc_pykt": 0.6051136363636365, "n_train_users": 6, "n_train_chunks": 6, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 23.7, "flip_max_diff": 0.0, "peak_gb": 0.96}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csNall-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csNall-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b824341543bd94516f38264e0a7e721cd6db7009
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csNall-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 786.6, "epochs": 53, "best_epoch": 43, "dev_auc_pykt": 0.7606813433665276, "n_train_users": 1730, "n_train_chunks": 2191, "n_dev_chunks": 525, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 48.6, "flip_max_diff": 0.0, "peak_gb": 9.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csNall-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csNall-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..272792301ed44e9a00785838f0fada2921d0ab97
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csNall-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1015.9, "epochs": 74, "best_epoch": 64, "dev_auc_pykt": 0.7610614821895909, "n_train_users": 1730, "n_train_chunks": 2173, "n_dev_chunks": 543, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 50.6, "flip_max_diff": 0.0, "peak_gb": 9.31}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csNall-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csNall-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..07cd72bb48b127387ab693669b847a3e17fe2919
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csNall-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 711.2, "epochs": 53, "best_epoch": 43, "dev_auc_pykt": 0.7623302159249152, "n_train_users": 1730, "n_train_chunks": 2164, "n_dev_chunks": 552, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 50.4, "flip_max_diff": 0.0, "peak_gb": 9.31}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csNall-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csNall-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..54a9840ae53a9c4007d46bb54e24badc1e628763
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csNall-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 835.6, "epochs": 56, "best_epoch": 46, "dev_auc_pykt": 0.7681778039920149, "n_train_users": 1730, "n_train_chunks": 2183, "n_dev_chunks": 533, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 56.2, "flip_max_diff": 0.0, "peak_gb": 9.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-csNall-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-csNall-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..bceb985734499d1d7a5b694805920c8116658315
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-csNall-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1095.0, "epochs": 74, "best_epoch": 64, "dev_auc_pykt": 0.766475544682264, "n_train_users": 1730, "n_train_chunks": 2151, "n_dev_chunks": 565, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 52.9, "flip_max_diff": 0.0, "peak_gb": 9.31}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-ics0/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-ics0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3c2a152631254b350ed944b9a601d61425179f44
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-ics0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 949.7, "epochs": 65, "best_epoch": 55, "dev_auc_pykt": 0.7597440087086109, "n_train_users": 1730, "n_train_chunks": 2066, "n_dev_chunks": 500, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 3.0, "flip_max_diff": 0.0, "peak_gb": 9.32}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-robustkt-ics1/train_info.json.done b/pykt/assist09/coldstart/pykt-robustkt-ics1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..9e5c767256cc861d20d72b9168ce095354412cdb
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-robustkt-ics1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1216.8, "epochs": 82, "best_epoch": 72, "dev_auc_pykt": 0.7529814999428891, "n_train_users": 1730, "n_train_chunks": 2051, "n_dev_chunks": 509, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 2.8, "flip_max_diff": 0.0, "peak_gb": 9.31}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-saint-ics0/train_info.json.done b/pykt/assist09/coldstart/pykt-saint-ics0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..aea0b303e4b660e23c206680e25f435b20974dbd
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-saint-ics0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 104.7, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.6691150082865884, "n_train_users": 1730, "n_train_chunks": 2066, "n_dev_chunks": 500, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 1.3, "flip_max_diff": 0.0, "peak_gb": 4.04}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-saint-ics1/train_info.json.done b/pykt/assist09/coldstart/pykt-saint-ics1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..f38db53890329a6bc54b83bef32d9ea347f9a045
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-saint-ics1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 81.0, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.6695667945040378, "n_train_users": 1730, "n_train_chunks": 2051, "n_dev_chunks": 509, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 1.0, "flip_max_diff": 0.0, "peak_gb": 4.04}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-saint-ics2/train_info.json.done b/pykt/assist09/coldstart/pykt-saint-ics2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b57f5ed137cec2dd6ca9364547d8603e2803ad20
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-saint-ics2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 90.2, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.6553980631779972, "n_train_users": 1730, "n_train_chunks": 2039, "n_dev_chunks": 518, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 1.3, "flip_max_diff": 0.0, "peak_gb": 4.04}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-saint_plus_plus-ics0/train_info.json.done b/pykt/assist09/coldstart/pykt-saint_plus_plus-ics0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..4a6ded91a2ff9d4a1689c3a59f3c776e2a04c836
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-saint_plus_plus-ics0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 227.3, "epochs": 31, "best_epoch": 21, "dev_auc_pykt": 0.6707089401739084, "n_train_users": 1730, "n_train_chunks": 2066, "n_dev_chunks": 500, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 1.3, "flip_max_diff": 0.0, "peak_gb": 4.28}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-saint_plus_plus-ics1/train_info.json.done b/pykt/assist09/coldstart/pykt-saint_plus_plus-ics1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..db654c59e0aaab2791ade5019a353e2566b59daa
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-saint_plus_plus-ics1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 133.5, "epochs": 20, "best_epoch": 10, "dev_auc_pykt": 0.6292301086255144, "n_train_users": 1730, "n_train_chunks": 2051, "n_dev_chunks": 509, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 1.4, "flip_max_diff": 0.0, "peak_gb": 4.28}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-saint_plus_plus-ics2/train_info.json.done b/pykt/assist09/coldstart/pykt-saint_plus_plus-ics2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e395927bc7a1d6ff508bcc262f5e7fd00231489b
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-saint_plus_plus-ics2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 96.4, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.6533815655041945, "n_train_users": 1730, "n_train_chunks": 2039, "n_dev_chunks": 518, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 1.1, "flip_max_diff": 0.0, "peak_gb": 4.28}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sakt-ics0/train_info.json.done b/pykt/assist09/coldstart/pykt-sakt-ics0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..36b65f025d9fa2addcbd94c4f3ec2e64f8867a24
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sakt-ics0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 21.4, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.7154420081678946, "n_train_users": 1730, "n_train_chunks": 2066, "n_dev_chunks": 500, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 0.6, "flip_max_diff": 0.0, "peak_gb": 0.42}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sakt-ics1/train_info.json.done b/pykt/assist09/coldstart/pykt-sakt-ics1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b58492a60b3a544b239de1ce931dc45f880995d7
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sakt-ics1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 22.0, "epochs": 17, "best_epoch": 7, "dev_auc_pykt": 0.7006658074850706, "n_train_users": 1730, "n_train_chunks": 2051, "n_dev_chunks": 509, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 0.3, "flip_max_diff": 0.0, "peak_gb": 0.42}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sakt-ics2/train_info.json.done b/pykt/assist09/coldstart/pykt-sakt-ics2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ac7bfc8d7a1f22f1197c5e3a9ccb3a069b51dec2
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sakt-ics2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 27.1, "epochs": 21, "best_epoch": 11, "dev_auc_pykt": 0.7168643610913665, "n_train_users": 1730, "n_train_chunks": 2039, "n_dev_chunks": 518, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 0.4, "flip_max_diff": 0.0, "peak_gb": 0.42}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN128-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN128-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..85f46206f0f0b5ad9d320a96e08b876b97eb7d3b
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN128-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 21.5, "epochs": 72, "best_epoch": 62, "dev_auc_pykt": 0.6936868764599857, "n_train_users": 102, "n_train_chunks": 122, "n_dev_chunks": 30, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 7.1, "flip_max_diff": 0.0, "peak_gb": 0.88}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN128-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN128-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..c9d6b59bb681f78da65c58a4166fc5c43c45148f
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN128-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 12.5, "epochs": 57, "best_epoch": 47, "dev_auc_pykt": 0.7202418100587435, "n_train_users": 102, "n_train_chunks": 127, "n_dev_chunks": 28, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 9.2, "flip_max_diff": 0.0, "peak_gb": 0.89}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN128-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN128-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..f4cd8f075be25ac44d69d2f500a9dcd71db16340
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN128-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 8.5, "epochs": 39, "best_epoch": 29, "dev_auc_pykt": 0.7488000030512437, "n_train_users": 102, "n_train_chunks": 116, "n_dev_chunks": 39, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 9.5, "flip_max_diff": 0.0, "peak_gb": 0.92}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN128-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN128-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e2a44dd81727159cac33684dc858f6961ef66407
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN128-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 12.1, "epochs": 46, "best_epoch": 36, "dev_auc_pykt": 0.749150159638521, "n_train_users": 102, "n_train_chunks": 130, "n_dev_chunks": 35, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 8.4, "flip_max_diff": 0.0, "peak_gb": 0.93}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN128-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN128-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7ad4acc3e2e71352cda5db777d33ddafd31f1a39
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN128-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 84.0, "epochs": 24, "best_epoch": 14, "dev_auc_pykt": 0.7353902808912599, "n_train_users": 102, "n_train_chunks": 119, "n_dev_chunks": 31, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 5.5, "flip_max_diff": 0.0, "peak_gb": 0.88}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN16-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN16-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..863b288a4b5254ad886424a0eed22edd16d9bb4d
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN16-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 5.1, "epochs": 24, "best_epoch": 14, "dev_auc_pykt": 0.5660038007329986, "n_train_users": 13, "n_train_chunks": 19, "n_dev_chunks": 4, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 18.2, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN16-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN16-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..f17e52753886148eed59c6b4b5aec34fef2b91c4
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN16-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 16.7, "epochs": 89, "best_epoch": 79, "dev_auc_pykt": 0.6280427223050175, "n_train_users": 13, "n_train_chunks": 18, "n_dev_chunks": 4, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 10.0, "flip_max_diff": 0.0, "peak_gb": 0.59}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN16-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN16-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6e93040a9e91914a8fdeb8581acb816ed7bdf9d4
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN16-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.9, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.4959156785243741, "n_train_users": 13, "n_train_chunks": 13, "n_dev_chunks": 3, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 9.3, "flip_max_diff": 0.0, "peak_gb": 0.58}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN16-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN16-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..dc3e51cf3367fc14c95dd38f6b37507569eb106c
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN16-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4.3, "epochs": 36, "best_epoch": 26, "dev_auc_pykt": 0.6359920459566947, "n_train_users": 13, "n_train_chunks": 16, "n_dev_chunks": 3, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 9.8, "flip_max_diff": 0.0, "peak_gb": 0.56}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN16-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN16-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a7c8b68ce2932de09cfa336601a89167fcad154f
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN16-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.5, "epochs": 28, "best_epoch": 18, "dev_auc_pykt": 0.58984375, "n_train_users": 13, "n_train_chunks": 15, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 9.4, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN256-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN256-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a5ba687f70ad54a6042ec80d02098707b6877ec7
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN256-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 13.3, "epochs": 40, "best_epoch": 30, "dev_auc_pykt": 0.7390871591578722, "n_train_users": 205, "n_train_chunks": 246, "n_dev_chunks": 66, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 9.2, "flip_max_diff": 0.0, "peak_gb": 0.92}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN256-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN256-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..49fd63a53b6db68bc10a84af661886fa8eaf5e87
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN256-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 24.2, "epochs": 51, "best_epoch": 41, "dev_auc_pykt": 0.7011416080363885, "n_train_users": 205, "n_train_chunks": 247, "n_dev_chunks": 64, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 9.8, "flip_max_diff": 0.0, "peak_gb": 0.9}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN256-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN256-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..21e6f5a22935eeb4d6ba85fa5f3b28370c29ef52
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN256-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 20.5, "epochs": 50, "best_epoch": 40, "dev_auc_pykt": 0.7612893150589118, "n_train_users": 205, "n_train_chunks": 261, "n_dev_chunks": 60, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 8.9, "flip_max_diff": 0.0, "peak_gb": 0.9}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN256-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN256-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..47efe8af21d196f5ff4d479ddf3c9e7c7c1026ce
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN256-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 27.3, "epochs": 61, "best_epoch": 51, "dev_auc_pykt": 0.703708741049515, "n_train_users": 205, "n_train_chunks": 265, "n_dev_chunks": 62, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 9.9, "flip_max_diff": 0.0, "peak_gb": 0.9}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN256-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN256-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7e9e9e5df4dacd31b7aaa8b7a8b1fe20fe3053a2
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN256-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 23.2, "epochs": 57, "best_epoch": 47, "dev_auc_pykt": 0.7185271405805032, "n_train_users": 205, "n_train_chunks": 245, "n_dev_chunks": 63, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 10.5, "flip_max_diff": 0.0, "peak_gb": 0.9}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN32-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN32-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..9bb5392158372504a58b6d3c8c840a82e67386f7
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN32-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3.5, "epochs": 34, "best_epoch": 24, "dev_auc_pykt": 0.6086412892470131, "n_train_users": 26, "n_train_chunks": 33, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 5.6, "flip_max_diff": 0.0, "peak_gb": 0.56}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN32-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN32-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8f3d0a267eeef3361fdd95752bbe0f3d0b79655f
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN32-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 5.0, "epochs": 30, "best_epoch": 20, "dev_auc_pykt": 0.7039927215595743, "n_train_users": 26, "n_train_chunks": 32, "n_dev_chunks": 10, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 7.4, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN32-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN32-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7bff696366901692433de0f13662991038e4ab75
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN32-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 7.6, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.8011874197689345, "n_train_users": 26, "n_train_chunks": 28, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 10.2, "flip_max_diff": 0.0, "peak_gb": 0.56}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN32-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN32-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..89ffafa519a5323066783575a17aab0bd7c12c00
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN32-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 10.7, "epochs": 23, "best_epoch": 13, "dev_auc_pykt": 0.5918435567441083, "n_train_users": 26, "n_train_chunks": 32, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 7.6, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN32-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN32-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..5193a346a2a0f63b2bb00be95918511ab414acaa
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN32-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 80.9, "epochs": 79, "best_epoch": 69, "dev_auc_pykt": 0.6426807760141093, "n_train_users": 26, "n_train_chunks": 31, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 6.6, "flip_max_diff": 0.0, "peak_gb": 0.58}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN64-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN64-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..0a644234f359d5b096be1a1e6d8e113126585ecd
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN64-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 13.4, "epochs": 49, "best_epoch": 39, "dev_auc_pykt": 0.6549954212454212, "n_train_users": 51, "n_train_chunks": 60, "n_dev_chunks": 16, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 11.6, "flip_max_diff": 0.0, "peak_gb": 0.83}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN64-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN64-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ec63d3e3f15d7782187a6663823bb6f7a32c7815
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN64-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.0, "epochs": 33, "best_epoch": 23, "dev_auc_pykt": 0.6453971548889186, "n_train_users": 51, "n_train_chunks": 65, "n_dev_chunks": 20, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 9.2, "flip_max_diff": 0.0, "peak_gb": 0.88}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN64-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN64-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..83b10688b86569286f34fe9fa4e7abbf8c6a5494
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN64-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 18.0, "epochs": 40, "best_epoch": 30, "dev_auc_pykt": 0.7082719365906855, "n_train_users": 51, "n_train_chunks": 59, "n_dev_chunks": 15, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 8.0, "flip_max_diff": 0.0, "peak_gb": 0.83}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN64-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN64-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1eac9213259b709afbe6b18f1adecb88f5bfc620
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN64-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 57.0, "epochs": 93, "best_epoch": 83, "dev_auc_pykt": 0.6774825174825174, "n_train_users": 51, "n_train_chunks": 59, "n_dev_chunks": 16, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 7.4, "flip_max_diff": 0.0, "peak_gb": 0.82}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN64-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN64-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ee44356390ebcde9c8f95655ede40e5d11564a31
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN64-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 149.5, "epochs": 27, "best_epoch": 17, "dev_auc_pykt": 0.704522932022932, "n_train_users": 51, "n_train_chunks": 58, "n_dev_chunks": 16, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 5.7, "flip_max_diff": 0.0, "peak_gb": 0.85}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN8-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN8-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..eee1924271e771cd8655db7a1532de8286e49cad
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN8-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 8.7, "epochs": 47, "best_epoch": 37, "dev_auc_pykt": 0.6053475935828877, "n_train_users": 6, "n_train_chunks": 11, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 14.9, "flip_max_diff": 0.0, "peak_gb": 0.58}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN8-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN8-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3f922ecf56fb2db7f0be38e63c6b7d723a45122e
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN8-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.6, "epochs": 33, "best_epoch": 23, "dev_auc_pykt": 0.6581286549707602, "n_train_users": 6, "n_train_chunks": 8, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 9.4, "flip_max_diff": 0.0, "peak_gb": 0.58}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN8-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN8-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e10818cd15b8d84db4e64d396289d4c21665ea58
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN8-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.6, "epochs": 16, "best_epoch": 6, "dev_auc_pykt": 0.5994986922406277, "n_train_users": 6, "n_train_chunks": 6, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 11.6, "flip_max_diff": 0.0, "peak_gb": 0.59}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN8-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN8-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6f5ac20ac1dd66bf59dbd464f6c808ff4a3952e4
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN8-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.1, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.64, "n_train_users": 6, "n_train_chunks": 7, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 9.9, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csN8-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csN8-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3d91b04e2f20eaf3a05278485485462f41c0b406
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csN8-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.7, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.5568181818181818, "n_train_users": 6, "n_train_chunks": 6, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 5.8, "flip_max_diff": 0.0, "peak_gb": 0.56}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csNall-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csNall-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..22d398b075a50dd9288bcd3053f3b5c218f090cf
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csNall-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 127.0, "epochs": 46, "best_epoch": 36, "dev_auc_pykt": 0.7636157762364474, "n_train_users": 1730, "n_train_chunks": 2191, "n_dev_chunks": 525, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 11.0, "flip_max_diff": 0.0, "peak_gb": 0.9}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csNall-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csNall-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..f48d4f244931b410cce4a015f628c791cb76511a
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csNall-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 126.6, "epochs": 46, "best_epoch": 36, "dev_auc_pykt": 0.7469978189107536, "n_train_users": 1730, "n_train_chunks": 2173, "n_dev_chunks": 543, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 10.7, "flip_max_diff": 0.0, "peak_gb": 0.91}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csNall-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csNall-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..d0be6deecdc0f91364050c1a1ebb0e88eb8f324e
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csNall-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 138.6, "epochs": 48, "best_epoch": 38, "dev_auc_pykt": 0.7615577616267388, "n_train_users": 1730, "n_train_chunks": 2164, "n_dev_chunks": 552, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 11.4, "flip_max_diff": 0.0, "peak_gb": 0.91}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csNall-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csNall-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..bd0040a4b4a5f20ae0fc4b2fdf4eb361088fd54d
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csNall-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 120.6, "epochs": 45, "best_epoch": 35, "dev_auc_pykt": 0.765106786469907, "n_train_users": 1730, "n_train_chunks": 2183, "n_dev_chunks": 533, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 9.8, "flip_max_diff": 0.0, "peak_gb": 0.9}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-sparsekt-csNall-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-sparsekt-csNall-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..20caa8cec0e529741df07487b5fcc68c81326b0a
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-sparsekt-csNall-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 124.7, "epochs": 45, "best_epoch": 35, "dev_auc_pykt": 0.7541556503940726, "n_train_users": 1730, "n_train_chunks": 2151, "n_dev_chunks": 565, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 10.3, "flip_max_diff": 0.0, "peak_gb": 0.9}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN128-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN128-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..19afd722045b8f34f98868bf4f7e1e8fc5fa25ed
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN128-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 16.4, "epochs": 34, "best_epoch": 24, "dev_auc_pykt": 0.6949720058963756, "n_train_users": 102, "n_train_chunks": 122, "n_dev_chunks": 30, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 18.6, "flip_max_diff": 0.0, "peak_gb": 2.99}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN128-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN128-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1618b0cad8fa49d0abe028b97306992d22454dea
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN128-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 13.9, "epochs": 35, "best_epoch": 25, "dev_auc_pykt": 0.7124357336027816, "n_train_users": 102, "n_train_chunks": 127, "n_dev_chunks": 28, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 17.2, "flip_max_diff": 0.0, "peak_gb": 2.99}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN128-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN128-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3bd50d166e192f0e5034465bb56c547d99b621e0
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN128-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 24.6, "epochs": 28, "best_epoch": 18, "dev_auc_pykt": 0.7541895801297922, "n_train_users": 102, "n_train_chunks": 116, "n_dev_chunks": 39, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 14.7, "flip_max_diff": 0.0, "peak_gb": 3.02}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN128-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN128-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6d810b0c4dec9bea34ac8158f8beb42f5fb766f8
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN128-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 15.3, "epochs": 30, "best_epoch": 20, "dev_auc_pykt": 0.7477921274456732, "n_train_users": 102, "n_train_chunks": 130, "n_dev_chunks": 35, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 16.1, "flip_max_diff": 0.0, "peak_gb": 2.99}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN128-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN128-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2350de13a54b56182db41b25dcb84f512531032d
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN128-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 38.3, "epochs": 20, "best_epoch": 10, "dev_auc_pykt": 0.7390295079279501, "n_train_users": 102, "n_train_chunks": 119, "n_dev_chunks": 31, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 17.2, "flip_max_diff": 0.0, "peak_gb": 3.02}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN16-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN16-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..613d12dbaef107b182587d4930e9e08d71ee84c8
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN16-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3.7, "epochs": 16, "best_epoch": 6, "dev_auc_pykt": 0.5619994570381431, "n_train_users": 13, "n_train_chunks": 19, "n_dev_chunks": 4, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 32.6, "flip_max_diff": 0.0, "peak_gb": 1.05}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN16-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN16-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..216c3c2b6518f9873fd235054873bfcc8f02b158
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN16-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 5.2, "epochs": 27, "best_epoch": 17, "dev_auc_pykt": 0.6465474416294088, "n_train_users": 13, "n_train_chunks": 18, "n_dev_chunks": 4, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 17.3, "flip_max_diff": 0.0, "peak_gb": 1.01}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN16-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN16-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8d1903223dd9392b6fd9f6346794cd73c47067d5
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN16-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 5.7, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6516469038208169, "n_train_users": 13, "n_train_chunks": 13, "n_dev_chunks": 3, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 16.6, "flip_max_diff": 0.0, "peak_gb": 0.81}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN16-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN16-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3bc0d9d35b5a470b9a238ee7beafb598b155d382
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN16-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.7, "epochs": 35, "best_epoch": 25, "dev_auc_pykt": 0.6226248342907645, "n_train_users": 13, "n_train_chunks": 16, "n_dev_chunks": 3, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 16.6, "flip_max_diff": 0.0, "peak_gb": 0.92}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN16-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN16-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..13da7b8328d95d61dea7f4713555ffbb0a096cdb
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN16-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 100.2, "epochs": 19, "best_epoch": 9, "dev_auc_pykt": 0.60166015625, "n_train_users": 13, "n_train_chunks": 15, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 16.1, "flip_max_diff": 0.0, "peak_gb": 0.88}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN256-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN256-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..356764bb1f0307667c555ea2c31d525f6e9b220e
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN256-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 18.8, "epochs": 21, "best_epoch": 11, "dev_auc_pykt": 0.7379673832355034, "n_train_users": 205, "n_train_chunks": 246, "n_dev_chunks": 66, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 16.9, "flip_max_diff": 0.0, "peak_gb": 2.99}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN256-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN256-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6a02bc63050f1ed46156c224241cd88651811468
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN256-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 21.3, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.68575041299724, "n_train_users": 205, "n_train_chunks": 247, "n_dev_chunks": 64, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 18.0, "flip_max_diff": 0.0, "peak_gb": 3.0}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN256-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN256-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1fc1f0b7b1609bf71077783bb6762c26b313f83b
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN256-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 15.3, "epochs": 21, "best_epoch": 11, "dev_auc_pykt": 0.7635154323834952, "n_train_users": 205, "n_train_chunks": 261, "n_dev_chunks": 60, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 14.2, "flip_max_diff": 0.0, "peak_gb": 2.99}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN256-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN256-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..679c355e2cb0df350628d3b2a5bc21edbfd9d387
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN256-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 55.5, "epochs": 21, "best_epoch": 11, "dev_auc_pykt": 0.6956924022605303, "n_train_users": 205, "n_train_chunks": 265, "n_dev_chunks": 62, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 19.3, "flip_max_diff": 0.0, "peak_gb": 3.0}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN256-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN256-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..5803a39cc68da8ed912cb51f484c469a162d4eff
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN256-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 19.9, "epochs": 32, "best_epoch": 22, "dev_auc_pykt": 0.7076416225776698, "n_train_users": 205, "n_train_chunks": 245, "n_dev_chunks": 63, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 16.1, "flip_max_diff": 0.0, "peak_gb": 2.99}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN32-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN32-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a4ebd0c4cdc59a29ad0fb9c372fa7aa77dcea1bd
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN32-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3.8, "epochs": 27, "best_epoch": 17, "dev_auc_pykt": 0.6135037510419561, "n_train_users": 26, "n_train_chunks": 33, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 11.0, "flip_max_diff": 0.0, "peak_gb": 1.65}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN32-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN32-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7e160291b2f1922086c36a35f0d5197463f263f4
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN32-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 5.4, "epochs": 27, "best_epoch": 17, "dev_auc_pykt": 0.7194794472232909, "n_train_users": 26, "n_train_chunks": 32, "n_dev_chunks": 10, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 13.5, "flip_max_diff": 0.0, "peak_gb": 1.63}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN32-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN32-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..79ea3b84d650b8978b6e950923ce8ad9b8705f84
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN32-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3.0, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.8096919127086007, "n_train_users": 26, "n_train_chunks": 28, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 14.6, "flip_max_diff": 0.0, "peak_gb": 1.44}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN32-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN32-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6842486ff344649cd9fe62f54819a4d7401fb80c
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN32-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.1, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.5986127360855759, "n_train_users": 26, "n_train_chunks": 32, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 10.1, "flip_max_diff": 0.0, "peak_gb": 1.61}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN32-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN32-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8df97ec110b6f907e0f2e1250105cb703c875078
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN32-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 62.5, "epochs": 59, "best_epoch": 49, "dev_auc_pykt": 0.6410934744268079, "n_train_users": 26, "n_train_chunks": 31, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 9.5, "flip_max_diff": 0.0, "peak_gb": 1.58}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN64-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN64-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..cdfcbd7a425592720dec2beccb435261c6c2ecf1
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN64-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 10.9, "epochs": 43, "best_epoch": 33, "dev_auc_pykt": 0.6548443223443223, "n_train_users": 51, "n_train_chunks": 60, "n_dev_chunks": 16, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 14.2, "flip_max_diff": 0.0, "peak_gb": 2.84}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN64-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN64-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..c222984891c7908ce45a2d26cef82bddf0187b6f
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN64-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 11.9, "epochs": 31, "best_epoch": 21, "dev_auc_pykt": 0.6593223144800083, "n_train_users": 51, "n_train_chunks": 65, "n_dev_chunks": 20, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 16.0, "flip_max_diff": 0.0, "peak_gb": 2.99}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN64-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN64-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..09c281c92627870c4db0afc8b751550a2b0e7449
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN64-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.4, "epochs": 34, "best_epoch": 24, "dev_auc_pykt": 0.7217144984972205, "n_train_users": 51, "n_train_chunks": 59, "n_dev_chunks": 15, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 13.9, "flip_max_diff": 0.0, "peak_gb": 2.78}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN64-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN64-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8271745d126a46bb937dd0ed05ef8144bd74bbe1
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN64-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 22.0, "epochs": 53, "best_epoch": 43, "dev_auc_pykt": 0.66380132687825, "n_train_users": 51, "n_train_chunks": 59, "n_dev_chunks": 16, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 11.6, "flip_max_diff": 0.0, "peak_gb": 2.78}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN64-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN64-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a1a3b7f9e9fef32af177b41f09dfdd900cd26bd8
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN64-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 7.2, "epochs": 28, "best_epoch": 18, "dev_auc_pykt": 0.7099959049959049, "n_train_users": 51, "n_train_chunks": 58, "n_dev_chunks": 16, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 9.9, "flip_max_diff": 0.0, "peak_gb": 2.79}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN8-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN8-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..c3e4abfad803fad4c29fdafee5803f1ffe8ef0d5
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN8-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 8.0, "epochs": 33, "best_epoch": 23, "dev_auc_pykt": 0.6294117647058824, "n_train_users": 6, "n_train_chunks": 11, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 30.3, "flip_max_diff": 0.0, "peak_gb": 0.71}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN8-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN8-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..29f233b2a70e6a90dcf5cb3f563a94970acaf0f1
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN8-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 14.5, "epochs": 32, "best_epoch": 22, "dev_auc_pykt": 0.6784795321637427, "n_train_users": 6, "n_train_chunks": 8, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 15.6, "flip_max_diff": 0.0, "peak_gb": 0.6}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN8-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN8-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..bd52422625f1fbd27a89e8dc0178ac5b21939242
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN8-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.8, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.6047297297297297, "n_train_users": 6, "n_train_chunks": 6, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 13.1, "flip_max_diff": 0.0, "peak_gb": 0.55}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN8-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN8-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b1b84da3c1e5cb84a6ecea758c6f9038237ea83f
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN8-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.5, "epochs": 40, "best_epoch": 30, "dev_auc_pykt": 0.6839999999999999, "n_train_users": 6, "n_train_chunks": 7, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 16.7, "flip_max_diff": 0.0, "peak_gb": 0.54}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csN8-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csN8-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2fa81245e66691b4b681afd6124ffdb52c11d6e7
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csN8-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.0, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.5965909090909092, "n_train_users": 6, "n_train_chunks": 6, "n_dev_chunks": 2, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 9.6, "flip_max_diff": 0.0, "peak_gb": 0.52}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csNall-s0/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csNall-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..abbe717b2f1e9f59a4d040255235f228a8a36df6
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csNall-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 240.8, "epochs": 35, "best_epoch": 25, "dev_auc_pykt": 0.7563637993808072, "n_train_users": 1730, "n_train_chunks": 2191, "n_dev_chunks": 525, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 18.9, "flip_max_diff": 0.0, "peak_gb": 2.99}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csNall-s1/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csNall-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..846961d59ddfcede6c71a21b989763aecf2978b7
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csNall-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 320.3, "epochs": 46, "best_epoch": 36, "dev_auc_pykt": 0.7367079164890964, "n_train_users": 1730, "n_train_chunks": 2173, "n_dev_chunks": 543, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 20.3, "flip_max_diff": 0.0, "peak_gb": 3.0}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csNall-s2/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csNall-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..d952b7d8941d03b5aac91facca53327d775ce01a
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csNall-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 336.6, "epochs": 47, "best_epoch": 37, "dev_auc_pykt": 0.7557565078332965, "n_train_users": 1730, "n_train_chunks": 2164, "n_dev_chunks": 552, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 20.1, "flip_max_diff": 0.0, "peak_gb": 3.0}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csNall-s3/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csNall-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..4554c8a02e7bc81335f05cf23f5d35e079d6023c
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csNall-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 295.7, "epochs": 40, "best_epoch": 30, "dev_auc_pykt": 0.7607821323462537, "n_train_users": 1730, "n_train_chunks": 2183, "n_dev_chunks": 533, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 14.5, "flip_max_diff": 0.0, "peak_gb": 2.99}
\ No newline at end of file
diff --git a/pykt/assist09/coldstart/pykt-stablekt-csNall-s4/train_info.json.done b/pykt/assist09/coldstart/pykt-stablekt-csNall-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..64580e72521e4ebb523c553dd7a02a1338fd0ace
--- /dev/null
+++ b/pykt/assist09/coldstart/pykt-stablekt-csNall-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 329.6, "epochs": 51, "best_epoch": 41, "dev_auc_pykt": 0.7535657638167674, "n_train_users": 1730, "n_train_chunks": 2151, "n_dev_chunks": 565, "batch_size": 64, "seq_len": 200, "n_val_users": 432, "score_s": 19.2, "flip_max_diff": 0.0, "peak_gb": 2.99}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-atdkt-s1/qid_model.ckpt b/pykt/assist09/pykt-atdkt-s1/qid_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..b8d4a600a0eb065ec48fd2dd26bc7cd9569300f4
--- /dev/null
+++ b/pykt/assist09/pykt-atdkt-s1/qid_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:127fcddadac155dd149c6cceb043547ca6bf5c2a41cef25bd1e89a82f3f0d715
+size 3443117
diff --git a/pykt/assist09/pykt-atdkt-s1/train_info.json b/pykt/assist09/pykt-atdkt-s1/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..bc3997b7d979258c2804bae9e970a1e7c3cc2fe6
--- /dev/null
+++ b/pykt/assist09/pykt-atdkt-s1/train_info.json
@@ -0,0 +1 @@
+{"train_s": 16.7, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.730859992708952, "n_train_users": 2162, "n_train_chunks": 2716, "n_dev_chunks": 331, "batch_size": 256, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "atdkt", "params": {"dataset_name": "algebra2005", "model_name": "atdkt", "emb_type": "qid", "save_dir": "saved_model", "seed": 3407, "fold": 0, "dropout": 0.05, "num_layers": 2, "num_attn_heads": 5, "l1": 0.5, "l2": 0.5, "l3": 0.5, "start": 50, "emb_size": 300, "learning_rate": 0.001, "use_wandb": 1, "add_uuid": 1}, "model_config": {"dropout": 0.05, "num_layers": 2, "num_attn_heads": 5, "l1": 0.5, "l3": 0.5, "start": 50, "emb_size": 300, "seq_len": 200}, "train_config": {"batch_size": 256, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "qid", "learning_rate": 0.001, "optimizer": {"name": "adam", "weight_decay": 0.0}, "patience": 10, "min_delta": 0.001, "level": "concept"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-atdkt-s2/qid_model.ckpt b/pykt/assist09/pykt-atdkt-s2/qid_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..ddbe69c300f813901a4a6f77f486a561e88a78dc
--- /dev/null
+++ b/pykt/assist09/pykt-atdkt-s2/qid_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ecc5fd14179fa7ceb7292b1af344a8d8b0492f6a50f0894efc79665ee37c462d
+size 3443117
diff --git a/pykt/assist09/pykt-atdkt-s2/train_info.json b/pykt/assist09/pykt-atdkt-s2/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..e6e1de421c078f541e0da981dc2a9861092a0771
--- /dev/null
+++ b/pykt/assist09/pykt-atdkt-s2/train_info.json
@@ -0,0 +1 @@
+{"train_s": 19.0, "epochs": 25, "best_epoch": 15, "dev_auc_pykt": 0.7313228232146769, "n_train_users": 2162, "n_train_chunks": 2716, "n_dev_chunks": 331, "batch_size": 256, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "atdkt", "params": {"dataset_name": "algebra2005", "model_name": "atdkt", "emb_type": "qid", "save_dir": "saved_model", "seed": 3407, "fold": 0, "dropout": 0.05, "num_layers": 2, "num_attn_heads": 5, "l1": 0.5, "l2": 0.5, "l3": 0.5, "start": 50, "emb_size": 300, "learning_rate": 0.001, "use_wandb": 1, "add_uuid": 1}, "model_config": {"dropout": 0.05, "num_layers": 2, "num_attn_heads": 5, "l1": 0.5, "l3": 0.5, "start": 50, "emb_size": 300, "seq_len": 200}, "train_config": {"batch_size": 256, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "qid", "learning_rate": 0.001, "optimizer": {"name": "adam", "weight_decay": 0.0}, "patience": 10, "min_delta": 0.001, "level": "concept"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-atktfix-s1/qid_model.ckpt b/pykt/assist09/pykt-atktfix-s1/qid_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..6e890f60c700f225ab829bb4a7adcc44f2a8dd8e
--- /dev/null
+++ b/pykt/assist09/pykt-atktfix-s1/qid_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a1458040e3d62267f3de3bb3769b5282b67007b51a70bad1d996ddac2bdc9b57
+size 793065
diff --git a/pykt/assist09/pykt-atktfix-s1/train_info.json b/pykt/assist09/pykt-atktfix-s1/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..4aedd5f4bff477724fab3724847a45a288620cd9
--- /dev/null
+++ b/pykt/assist09/pykt-atktfix-s1/train_info.json
@@ -0,0 +1 @@
+{"train_s": 230.1, "epochs": 97, "best_epoch": 87, "dev_auc_pykt": 0.713642682029318, "n_train_users": 2162, "n_train_chunks": 2716, "n_dev_chunks": 331, "batch_size": 256, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "atktfix", "params": {"dataset_name": "assist2015", "model_name": "atktfix", "emb_type": "qid", "save_dir": "saved_model", "seed": 42, "fold": 0, "dropout": 0.2, "skill_dim": 256, "answer_dim": 96, "hidden_dim": 80, "attention_dim": 80, "epsilon": 10, "beta": 0.2, "learning_rate": 0.001, "use_wandb": 1, "add_uuid": 1}, "model_config": {"dropout": 0.2, "skill_dim": 256, "answer_dim": 96, "hidden_dim": 80, "attention_dim": 80, "epsilon": 10, "beta": 0.2}, "train_config": {"batch_size": 256, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "qid", "learning_rate": 0.001, "optimizer": {"name": "adam", "weight_decay": 0.0}, "patience": 10, "min_delta": 0.001, "level": "concept"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-atktfix-s2/qid_model.ckpt b/pykt/assist09/pykt-atktfix-s2/qid_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..b57a3497a0d2001c2bbbaee4295916668be0eb49
--- /dev/null
+++ b/pykt/assist09/pykt-atktfix-s2/qid_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:237dc5caffb0b7904d51fd3c2605e50de508b1bab537744f6edb9d5b67f0edc9
+size 793065
diff --git a/pykt/assist09/pykt-atktfix-s2/train_info.json b/pykt/assist09/pykt-atktfix-s2/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..1a5b933979ab10b04bc57597a245929a034523e9
--- /dev/null
+++ b/pykt/assist09/pykt-atktfix-s2/train_info.json
@@ -0,0 +1 @@
+{"train_s": 178.5, "epochs": 87, "best_epoch": 77, "dev_auc_pykt": 0.7105722292303501, "n_train_users": 2162, "n_train_chunks": 2716, "n_dev_chunks": 331, "batch_size": 256, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "atktfix", "params": {"dataset_name": "assist2015", "model_name": "atktfix", "emb_type": "qid", "save_dir": "saved_model", "seed": 42, "fold": 0, "dropout": 0.2, "skill_dim": 256, "answer_dim": 96, "hidden_dim": 80, "attention_dim": 80, "epsilon": 10, "beta": 0.2, "learning_rate": 0.001, "use_wandb": 1, "add_uuid": 1}, "model_config": {"dropout": 0.2, "skill_dim": 256, "answer_dim": 96, "hidden_dim": 80, "attention_dim": 80, "epsilon": 10, "beta": 0.2}, "train_config": {"batch_size": 256, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "qid", "learning_rate": 0.001, "optimizer": {"name": "adam", "weight_decay": 0.0}, "patience": 10, "min_delta": 0.001, "level": "concept"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-dimkt-s1/qid_model.ckpt b/pykt/assist09/pykt-dimkt-s1/qid_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..eea773764f5b691100546be211b4cf8ba513f076
--- /dev/null
+++ b/pykt/assist09/pykt-dimkt-s1/qid_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4d2e4055d78c4b68a333b827c9184d147427820242802e86c49b9174cb725179
+size 10288476
diff --git a/pykt/assist09/pykt-dimkt-s1/train_info.json b/pykt/assist09/pykt-dimkt-s1/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..2687bc9bf46dbfc8c578036901bc97f00bdb21ec
--- /dev/null
+++ b/pykt/assist09/pykt-dimkt-s1/train_info.json
@@ -0,0 +1 @@
+{"train_s": 502.6, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.7408527302208106, "n_train_users": 2162, "n_train_chunks": 2716, "n_dev_chunks": 331, "batch_size": 64, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "dimkt", "params": {"dataset_name": "assist2012", "model_name": "dimkt", "emb_type": "qid", "save_dir": "saved_model", "learning_rate": 0.0005, "seed": 42, "fold": 0, "weight_decay": 0.0001, "dropout": 0.2, "emb_size": 128, "batch_size": 64, "num_steps": 199, "difficult_levels": 100, "use_wandb": 1, "add_uuid": 1}, "model_config": {"dropout": 0.2, "emb_size": 128, "num_steps": 199, "difficult_levels": 100, "batch_size": 64}, "train_config": {"batch_size": 64, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "qid", "learning_rate": 0.0005, "optimizer": {"name": "adam", "weight_decay": 0.0001}, "patience": 10, "min_delta": 0.001, "level": "concept"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-dimkt-s2/qid_model.ckpt b/pykt/assist09/pykt-dimkt-s2/qid_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..a46639f3fb5862ffa0330f31e555164b3ed47b13
--- /dev/null
+++ b/pykt/assist09/pykt-dimkt-s2/qid_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ecee137a6b049396bb1e8aa7661996eeb82e01eb489c82dddf2e7dab80bb7c0
+size 10288476
diff --git a/pykt/assist09/pykt-dimkt-s2/train_info.json b/pykt/assist09/pykt-dimkt-s2/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..241b1eb9b562f066a2bab16ca88719ca53165547
--- /dev/null
+++ b/pykt/assist09/pykt-dimkt-s2/train_info.json
@@ -0,0 +1 @@
+{"train_s": 477.8, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.7372337986792337, "n_train_users": 2162, "n_train_chunks": 2716, "n_dev_chunks": 331, "batch_size": 64, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "dimkt", "params": {"dataset_name": "assist2012", "model_name": "dimkt", "emb_type": "qid", "save_dir": "saved_model", "learning_rate": 0.0005, "seed": 42, "fold": 0, "weight_decay": 0.0001, "dropout": 0.2, "emb_size": 128, "batch_size": 64, "num_steps": 199, "difficult_levels": 100, "use_wandb": 1, "add_uuid": 1}, "model_config": {"dropout": 0.2, "emb_size": 128, "num_steps": 199, "difficult_levels": 100, "batch_size": 64}, "train_config": {"batch_size": 64, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "qid", "learning_rate": 0.0005, "optimizer": {"name": "adam", "weight_decay": 0.0001}, "patience": 10, "min_delta": 0.001, "level": "concept"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-dtransformer-s1/qid_cl_model.ckpt b/pykt/assist09/pykt-dtransformer-s1/qid_cl_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..90c6e73ba859a683a5f594e83d4b77a83b2364dd
--- /dev/null
+++ b/pykt/assist09/pykt-dtransformer-s1/qid_cl_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d36bc6f9e8345afedfd7055d608762332fd08378840ee705da475c7055430b52
+size 4714409
diff --git a/pykt/assist09/pykt-dtransformer-s1/train_info.json b/pykt/assist09/pykt-dtransformer-s1/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..975abbf3a81246641bf20765809fff70ee7485c0
--- /dev/null
+++ b/pykt/assist09/pykt-dtransformer-s1/train_info.json
@@ -0,0 +1 @@
+{"train_s": 1668.6, "epochs": 29, "best_epoch": 19, "dev_auc_pykt": 0.7500008663196986, "n_train_users": 2162, "n_train_chunks": 2716, "n_dev_chunks": 331, "batch_size": 32, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "dtransformer", "params": {"dataset_name": "statics2011", "model_name": "dtransformer", "emb_type": "qid_cl", "save_dir": "saved_model", "seed": 3407, "fold": 0, "dropout": 0.3, "d_model": 256, "d_ff": 256, "num_attn_heads": 8, "n_blocks": 4, "learning_rate": 0.001, "n_know": 16, "lambda_cl": 0.1, "window": 1, "proj": true, "hard_neg": false, "use_wandb": 1, "add_uuid": 1}, "model_config": {"dropout": 0.3, "d_model": 256, "d_ff": 256, "num_attn_heads": 8, "n_blocks": 4, "n_know": 16, "lambda_cl": 0.1, "window": 1, "proj": true, "hard_neg": false}, "train_config": {"batch_size": 32, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "qid_cl", "learning_rate": 0.001, "optimizer": {"name": "adam", "weight_decay": 1e-05}, "patience": 10, "min_delta": 0.001, "level": "concept"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-iekt-s1/qid_model.ckpt b/pykt/assist09/pykt-iekt-s1/qid_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..32bcb56780d7082d740e30e8f61ce6f0828382fb
--- /dev/null
+++ b/pykt/assist09/pykt-iekt-s1/qid_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aeb0ed0c5642da578e47a4766bd1a45316bfb7b20436dca6133db0975ec7c43d
+size 12392847
diff --git a/pykt/assist09/pykt-iekt-s1/train_info.json b/pykt/assist09/pykt-iekt-s1/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..6488ef696318bbf2c6825d5b59746fb41fb5698b
--- /dev/null
+++ b/pykt/assist09/pykt-iekt-s1/train_info.json
@@ -0,0 +1 @@
+{"train_s": 2568.9, "epochs": 70, "best_epoch": 60, "dev_auc_pykt": 0.7545279337083168, "n_train_users": 2162, "n_train_chunks": 2571, "n_dev_chunks": 310, "batch_size": 256, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "iekt", "params": {"dataset_name": "assist2009", "model_name": "iekt", "emb_type": "qid", "save_dir": "saved_model", "learning_rate": 0.001, "seed": 42, "fold": 0, "dropout": 0, "emb_size": 64, "lamb": 40, "n_layer": 1, "cog_levels": 10, "acq_levels": 10, "gamma": 0.93, "use_wandb": 1, "add_uuid": 1}, "model_config": {"dropout": 0, "emb_size": 64, "lamb": 40, "n_layer": 1, "cog_levels": 10, "acq_levels": 10, "gamma": 0.93}, "train_config": {"batch_size": 256, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "qid", "learning_rate": 0.001, "optimizer": {"name": "adam", "weight_decay": 1e-06}, "patience": 10, "min_delta": 0.001, "level": "question"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-iekt-s2/qid_model.ckpt b/pykt/assist09/pykt-iekt-s2/qid_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..315a6c5b72c6b78b9b2669e173d2ba3a1dd9e777
--- /dev/null
+++ b/pykt/assist09/pykt-iekt-s2/qid_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a6f56e4ef4226305ff0a731439e33cfe6fe63cb4fb3b3c4f74b6ce282aad0a61
+size 12392847
diff --git a/pykt/assist09/pykt-iekt-s2/train_info.json b/pykt/assist09/pykt-iekt-s2/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..bbb420d0681267e4a0d9bf28d03d575cbdf16020
--- /dev/null
+++ b/pykt/assist09/pykt-iekt-s2/train_info.json
@@ -0,0 +1 @@
+{"train_s": 3447.0, "epochs": 91, "best_epoch": 81, "dev_auc_pykt": 0.7612506013538745, "n_train_users": 2162, "n_train_chunks": 2571, "n_dev_chunks": 310, "batch_size": 256, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "iekt", "params": {"dataset_name": "assist2009", "model_name": "iekt", "emb_type": "qid", "save_dir": "saved_model", "learning_rate": 0.001, "seed": 42, "fold": 0, "dropout": 0, "emb_size": 64, "lamb": 40, "n_layer": 1, "cog_levels": 10, "acq_levels": 10, "gamma": 0.93, "use_wandb": 1, "add_uuid": 1}, "model_config": {"dropout": 0, "emb_size": 64, "lamb": 40, "n_layer": 1, "cog_levels": 10, "acq_levels": 10, "gamma": 0.93}, "train_config": {"batch_size": 256, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "qid", "learning_rate": 0.001, "optimizer": {"name": "adam", "weight_decay": 1e-06}, "patience": 10, "min_delta": 0.001, "level": "question"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-mockt-s1/qid_conv_ker_noexp_model.ckpt b/pykt/assist09/pykt-mockt-s1/qid_conv_ker_noexp_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..7380d11c92357756b079a0140a9d768130b47f35
--- /dev/null
+++ b/pykt/assist09/pykt-mockt-s1/qid_conv_ker_noexp_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:47b43972288fa0956fe544aca18a7e5be040d968e1663f44d3f129e7055615d5
+size 16511139
diff --git a/pykt/assist09/pykt-mockt-s1/train_info.json b/pykt/assist09/pykt-mockt-s1/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..3219e92b490012e110af4e81949082b40278e6da
--- /dev/null
+++ b/pykt/assist09/pykt-mockt-s1/train_info.json
@@ -0,0 +1 @@
+{"train_s": 593.9, "epochs": 115, "best_epoch": 105, "dev_auc_pykt": 0.7605957499688378, "n_train_users": 2162, "n_train_chunks": 2716, "n_dev_chunks": 331, "batch_size": 256, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "mockt", "params": {"dataset_name": "bridge2algebra2006", "model_name": "mockt", "emb_type": "qid_conv_ker_noexp", "save_dir": "saved_model", "seed": 3407, "fold": 0, "dropout": 0.1, "d_model": 256, "d_ff": 64, "num_attn_heads": 4, "n_blocks": 4, "learning_rate": 0.0001, "num_buckets": 256, "max_distance": 256, "bar_d": 64, "max_local_shift": 0.0, "n_hat": 0, "d_state": 16, "d_conv": 2, "expand": 1, "kernel_size": 4, "sequence1": 50, "sequence2": 150, "use_wandb": 0, "add_uuid": 0}, "model_config": {"dropout": 0.1, "d_model": 256, "d_ff": 64, "num_attn_heads": 4, "n_blocks": 4, "num_buckets": 256, "max_distance": 256, "bar_d": 64, "max_local_shift": 0.0, "n_hat": 0, "d_state": 16, "d_conv": 2, "expand": 1, "kernel_size": 4, "sequence1": 50, "sequence2": 150}, "train_config": {"batch_size": 256, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "qid_conv_ker_noexp", "learning_rate": 0.0001, "optimizer": {"name": "adam", "weight_decay": 0.0}, "patience": 10, "min_delta": 0.001, "level": "concept"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-qdkt-s1/qid_model.ckpt b/pykt/assist09/pykt-qdkt-s1/qid_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..c857e36780fd784167ef38f607b46c7d927b6acb
--- /dev/null
+++ b/pykt/assist09/pykt-qdkt-s1/qid_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2897d22d98edde2ed4fb704730c11f7c724764106a25938370b3b4f9313cb37d
+size 66701297
diff --git a/pykt/assist09/pykt-qdkt-s1/train_info.json b/pykt/assist09/pykt-qdkt-s1/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..aae225d4ab3a57a30c3756cdb42d0758584f0778
--- /dev/null
+++ b/pykt/assist09/pykt-qdkt-s1/train_info.json
@@ -0,0 +1 @@
+{"train_s": 174.3, "epochs": 40, "best_epoch": 30, "dev_auc_pykt": 0.6959265986381319, "n_train_users": 2162, "n_train_chunks": 2571, "n_dev_chunks": 310, "batch_size": 32, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "qdkt", "params": {"dataset_name": "assist2009", "model_name": "qdkt", "emb_type": "qid", "save_dir": "saved_model", "seed": 42, "fold": 0, "dropout": 0.2, "emb_size": 300, "learning_rate": 0.001, "batch_size": 32, "num_epochs": 200, "use_wandb": 1, "add_uuid": 1}, "model_config": {"dropout": 0.2, "emb_size": 300}, "train_config": {"batch_size": 32, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "qid", "learning_rate": 0.001, "optimizer": {"name": "adam", "weight_decay": 0.0}, "patience": 10, "min_delta": 0.001, "level": "question"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-qdkt-s2/qid_model.ckpt b/pykt/assist09/pykt-qdkt-s2/qid_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..cd4ad77690b6f4a852d735b1d1491ead23e3fc85
--- /dev/null
+++ b/pykt/assist09/pykt-qdkt-s2/qid_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:edf8b468cbb5ee8ddee24f9e6009953212602ddbeb3df267fc8861ac6ab8f9d8
+size 66701297
diff --git a/pykt/assist09/pykt-qdkt-s2/train_info.json b/pykt/assist09/pykt-qdkt-s2/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..cce9ae0243978ba866e4dfe6871d8fdee6e2397a
--- /dev/null
+++ b/pykt/assist09/pykt-qdkt-s2/train_info.json
@@ -0,0 +1 @@
+{"train_s": 136.3, "epochs": 33, "best_epoch": 23, "dev_auc_pykt": 0.6950995989234929, "n_train_users": 2162, "n_train_chunks": 2571, "n_dev_chunks": 310, "batch_size": 32, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "qdkt", "params": {"dataset_name": "assist2009", "model_name": "qdkt", "emb_type": "qid", "save_dir": "saved_model", "seed": 42, "fold": 0, "dropout": 0.2, "emb_size": 300, "learning_rate": 0.001, "batch_size": 32, "num_epochs": 200, "use_wandb": 1, "add_uuid": 1}, "model_config": {"dropout": 0.2, "emb_size": 300}, "train_config": {"batch_size": 32, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "qid", "learning_rate": 0.001, "optimizer": {"name": "adam", "weight_decay": 0.0}, "patience": 10, "min_delta": 0.001, "level": "question"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-qikt-s1/iekt_model.ckpt b/pykt/assist09/pykt-qikt-s1/iekt_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..9e2c4b38d6809396915cbe137d5bd5bebf5f4feb
--- /dev/null
+++ b/pykt/assist09/pykt-qikt-s1/iekt_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f28bd357256422edd8563298fd0c4d4604fa9d2743a9550777a467f076ec1cfa
+size 72886645
diff --git a/pykt/assist09/pykt-qikt-s1/train_info.json b/pykt/assist09/pykt-qikt-s1/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..b3750af2aeb3b37a0e172a7132479bed783710cc
--- /dev/null
+++ b/pykt/assist09/pykt-qikt-s1/train_info.json
@@ -0,0 +1 @@
+{"train_s": 120.9, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.750735075597775, "n_train_users": 2162, "n_train_chunks": 2571, "n_dev_chunks": 310, "batch_size": 32, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "qikt", "params": {"dataset_name": "assist2009", "fold": 0, "learning_rate": 0.001, "seed": 3407, "batch_size": 32, "num_epochs": 200, "use_wandb": 1, "add_uuid": 1, "save_dir": "saved_model", "model_name": "qikt", "emb_type": "iekt", "dropout": 0.4, "emb_size": 300, "mlp_layer_num": 2, "loss_q_all_lambda": 0, "loss_c_all_lambda": 0, "loss_q_next_lambda": 0, "loss_c_next_lambda": 0, "output_q_all_lambda": 1, "output_c_all_lambda": 1, "output_q_next_lambda": 0, "output_c_next_lambda": 1, "output_mode": "an"}, "model_config": {"dropout": 0.4, "emb_size": 300, "mlp_layer_num": 2, "loss_q_all_lambda": 0, "loss_c_all_lambda": 0, "loss_q_next_lambda": 0, "loss_c_next_lambda": 0, "output_q_all_lambda": 1, "output_c_all_lambda": 1, "output_q_next_lambda": 0, "output_c_next_lambda": 1, "output_mode": "an"}, "train_config": {"batch_size": 32, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "iekt", "learning_rate": 0.001, "optimizer": {"name": "adam", "weight_decay": 0.0}, "patience": 10, "min_delta": 0.001, "level": "question"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist09/pykt-qikt-s2/iekt_model.ckpt b/pykt/assist09/pykt-qikt-s2/iekt_model.ckpt
new file mode 100644
index 0000000000000000000000000000000000000000..a7eb009173f47e987b7c25649532d9472cf4abc7
--- /dev/null
+++ b/pykt/assist09/pykt-qikt-s2/iekt_model.ckpt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:14cfbb0ded273b682c14c61f4f2ee41fbb30fa28afb0b4fba9fc772ccafabd49
+size 72886645
diff --git a/pykt/assist09/pykt-qikt-s2/train_info.json b/pykt/assist09/pykt-qikt-s2/train_info.json
new file mode 100644
index 0000000000000000000000000000000000000000..c2a468d6df8e315727d9f2a8ca533f1594a1ccfd
--- /dev/null
+++ b/pykt/assist09/pykt-qikt-s2/train_info.json
@@ -0,0 +1 @@
+{"train_s": 124.1, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.7519052331674958, "n_train_users": 2162, "n_train_chunks": 2571, "n_dev_chunks": 310, "batch_size": 32, "seq_len": 200, "n_val_users": 240, "cfg": {"model": "qikt", "params": {"dataset_name": "assist2009", "fold": 0, "learning_rate": 0.001, "seed": 3407, "batch_size": 32, "num_epochs": 200, "use_wandb": 1, "add_uuid": 1, "save_dir": "saved_model", "model_name": "qikt", "emb_type": "iekt", "dropout": 0.4, "emb_size": 300, "mlp_layer_num": 2, "loss_q_all_lambda": 0, "loss_c_all_lambda": 0, "loss_q_next_lambda": 0, "loss_c_next_lambda": 0, "output_q_all_lambda": 1, "output_c_all_lambda": 1, "output_q_next_lambda": 0, "output_c_next_lambda": 1, "output_mode": "an"}, "model_config": {"dropout": 0.4, "emb_size": 300, "mlp_layer_num": 2, "loss_q_all_lambda": 0, "loss_c_all_lambda": 0, "loss_q_next_lambda": 0, "loss_c_next_lambda": 0, "output_q_all_lambda": 1, "output_c_all_lambda": 1, "output_q_next_lambda": 0, "output_c_next_lambda": 1, "output_mode": "an"}, "train_config": {"batch_size": 32, "num_epochs": 200, "optimizer": "adam", "seq_len": 200}, "emb_type": "iekt", "learning_rate": 0.001, "optimizer": {"name": "adam", "weight_decay": 0.0}, "patience": 10, "min_delta": 0.001, "level": "question"}, "users": [14, 21825, 53167, 54318, 64525, 64531, 64532, 64535, 64550, 64634, 70677, 70679, 70682, 70684, 70687, 70688, 70693, 70695, 70699, 70702, 70704, 70706, 70708, 70709, 70728, 70729, 70730, 70732, 70733, 70736, 70740, 70743, 70745, 70746, 70752, 70754, 70758, 70759, 70816, 70874, 71034, 71066, 71068, 71105, 71148, 71151, 71152, 71173, 71174, 71176, 71180, 71183, 71188, 71190, 71193, 71203, 71208, 71209, 71212, 71213, 71219, 71224, 71227, 71230, 71231, 71232, 71235, 71427, 71564, 71740, 71807, 71809, 71810, 71823, 71824, 71826, 71829, 71831, 71872, 71873, 71874, 71876, 71877, 71881, 71883, 71907, 71918, 72059, 72327, 72378, 72720, 73100, 73604, 73684, 73685, 73962, 74384, 74459, 74673, 74674, 74676, 74698, 74704, 74964, 75316, 75346, 76685, 76992, 76998, 77069, 77118, 77866, 77881, 77902, 77939, 77942, 77944, 77945, 77948, 77955, 77956, 77957, 77958, 77960, 77962, 77966, 77967, 77973, 77990, 77991, 77993, 77997, 77998, 78000, 78001, 78004, 78005, 78006, 78007, 78012, 78016, 78017, 78018, 78019, 78021, 78022, 78026, 78028, 78029, 78030, 78031, 78032, 78033, 78034, 78035, 78037, 78038, 78039, 78040, 78041, 78042, 78043, 78044, 78045, 78049, 78051, 78052, 78061, 78062, 78090, 78091, 78092, 78093, 78094, 78097, 78099, 78100, 78102, 78103, 78105, 78106, 78110, 78137, 78157, 78160, 78161, 78163, 78166, 78167, 78168, 78174, 78175, 78178, 78184, 78202, 78204, 78206, 78207, 78208, 78210, 78211, 78212, 78213, 78214, 78216, 78217, 78218, 78220, 78228, 78256, 78265, 78268, 78270, 78272, 78273, 78274, 78275, 78277, 78278, 78279, 78280, 78281, 78284, 78285, 78286, 78340, 78341, 78344, 78345, 78351, 78353, 78356, 78357, 78361, 78362, 78363, 78365, 78366, 78367, 78368, 78369, 78370, 78371, 78389, 78392, 78394, 78397, 78398, 78405, 78406, 78409, 78411, 78412, 78413, 78414, 78415, 78418, 78419, 78423, 78447, 78449, 78450, 78452, 78453, 78454, 78456, 78459, 78460, 78461, 78492, 78493, 78495, 78498, 78499, 78500, 78501, 78503, 78504, 78509, 78510, 78512, 78513, 78514, 78517, 78519, 78520, 78521, 78522, 78523, 78527, 78531, 78534, 78537, 78539, 78540, 78541, 78542, 78543, 78545, 78546, 78548, 78549, 78551, 78552, 78556, 78557, 78558, 78559, 78560, 78565, 78567, 78568, 78570, 78571, 78574, 78579, 78581, 78582, 78583, 78584, 78590, 78591, 78593, 78647, 78674, 78675, 78705, 78706, 78708, 78711, 78712, 78718, 78719, 78721, 78722, 78725, 78729, 78763, 78764, 78765, 78766, 78767, 78769, 78770, 78771, 78772, 78773, 78776, 78778, 78779, 78780, 78781, 78782, 78787, 78788, 78789, 78803, 78804, 78805, 78806, 78807, 78809, 78811, 78812, 78814, 78816, 78817, 78819, 78833, 78835, 78836, 78837, 78838, 78839, 78840, 78842, 78843, 78845, 78846, 78847, 78848, 78849, 78863, 78875, 78877, 78878, 78879, 78880, 78882, 78883, 78884, 78887, 78888, 78891, 78892, 78894, 78895, 78897, 78899, 78901, 78903, 78905, 78906, 78908, 78909, 78910, 78911, 78912, 78913, 78914, 78916, 78918, 78919, 78920, 78923, 78927, 78944, 78945, 78947, 78948, 78949, 78951, 78952, 78954, 78955, 78957, 78958, 78959, 78968, 78969, 78971, 78972, 78973, 78974, 78975, 78976, 78977, 78978, 78979, 78982, 78983, 78984, 78986, 78987, 78990, 78991, 78994, 78995, 78997, 79000, 79001, 79002, 79003, 79005, 79006, 79010, 79011, 79012, 79014, 79015, 79017, 79018, 79020, 79021, 79023, 79024, 79027, 79029, 79031, 79032, 79034, 79054, 79055, 79056, 79058, 79059, 79060, 79061, 79062, 79064, 79067, 79068, 79070, 79071, 79073, 79074, 79111, 79148, 79150, 79156, 79158, 79161, 79213, 79217, 79218, 79219, 79220, 79224, 79237, 79242, 79246, 79264, 79272, 79276, 79281, 79291, 79300, 79316, 79319, 79324, 79327, 79403, 79409, 79411, 79412, 79413, 79414, 79415, 79416, 79417, 79421, 79422, 79425, 79426, 79428, 79430, 79432, 79448, 79451, 79458, 79460, 79481, 79482, 79484, 79485, 79487, 79488, 79490, 79491, 79492, 79493, 79501, 79503, 79506, 79508, 79509, 79510, 79511, 79513, 79517, 79518, 79519, 79522, 79528, 79529, 79530, 79531, 79534, 79535, 79537, 79538, 79539, 79540, 79545, 79546, 79548, 79555, 79558, 79559, 79561, 79564, 79568, 79572, 79577, 79578, 79581, 79582, 79583, 79584, 79585, 79586, 79588, 79595, 79596, 79597, 79598, 79602, 79605, 79606, 79608, 79609, 79610, 79611, 79613, 79614, 79615, 79616, 79618, 79620, 79621, 79622, 79623, 79625, 79632, 79661, 79662, 79664, 79665, 79666, 79668, 79669, 79670, 79671, 79672, 79674, 79675, 79678, 79680, 79681, 79684, 79686, 79688, 79689, 79690, 79691, 79692, 79694, 79695, 79697, 79699, 79700, 79704, 79714, 79716, 79720, 79721, 79726, 79729, 79735, 79736, 79737, 79738, 79740, 79741, 79742, 79743, 79744, 79745, 79747, 79748, 79749, 79750, 79760, 79761, 79762, 79763, 79764, 79765, 79766, 79768, 79769, 79773, 79774, 79776, 79777, 79797, 79803, 79804, 79831, 79833, 79857, 79865, 79884, 79896, 79908, 79911, 79962, 79963, 79990, 80008, 80009, 80010, 80011, 80012, 80014, 80016, 80018, 80019, 80022, 80023, 80051, 80052, 80087, 80088, 80089, 80090, 80095, 80109, 80114, 80115, 80117, 80119, 80120, 80121, 80123, 80124, 80125, 80126, 80127, 80128, 80129, 80130, 80132, 80133, 80135, 80136, 80137, 80138, 80139, 80144, 80147, 80150, 80160, 80161, 80169, 80171, 80172, 80173, 80174, 80175, 80180, 80182, 80210, 80213, 80218, 80223, 80224, 80225, 80226, 80227, 80228, 80229, 80231, 80234, 80238, 80243, 80244, 80247, 80248, 80249, 80250, 80254, 80255, 80256, 80350, 80352, 80353, 80354, 80356, 80357, 80358, 80360, 80449, 80506, 80612, 80613, 80614, 80615, 80616, 80625, 80633, 80634, 80635, 80636, 80637, 80638, 80639, 80640, 80641, 80642, 80644, 80646, 80648, 80649, 80650, 80651, 80652, 80653, 80654, 80656, 80657, 80658, 80659, 80660, 80661, 80663, 80664, 80665, 80715, 80717, 80718, 80720, 80771, 80807, 80904, 80906, 80911, 80912, 80914, 80915, 80918, 80919, 80923, 80928, 80984, 80988, 80989, 80996, 81011, 81013, 81020, 81028, 81037, 81040, 81054, 81081, 81083, 81084, 81086, 81096, 81179, 81192, 81195, 81198, 81200, 81211, 81220, 81223, 81225, 81231, 81237, 81254, 81297, 81309, 81319, 81347, 81363, 81390, 81405, 81407, 81435, 81439, 81441, 81442, 81443, 81445, 81446, 81447, 81449, 81450, 81451, 81452, 81454, 81455, 81457, 81458, 81460, 81461, 81464, 81468, 81485, 81488, 81541, 81567, 81577, 81578, 81604, 81618, 81619, 81623, 81624, 81626, 81627, 81628, 81630, 81641, 81646, 81648, 81766, 81837, 81887, 81888, 81896, 81903, 81904, 81905, 81928, 81930, 81935, 81937, 81944, 81946, 81947, 81981, 81983, 81984, 82012, 82022, 82023, 82025, 82029, 82030, 82031, 82032, 82037, 82046, 82047, 82054, 82056, 82057, 82058, 82065, 82079, 82081, 82083, 82088, 82089, 82090, 82091, 82092, 82093, 82105, 82107, 82142, 82151, 82155, 82162, 82163, 82207, 82241, 82250, 82252, 82271, 82277, 82279, 82291, 82293, 82296, 82297, 82301, 82303, 82325, 82333, 82335, 82336, 82340, 82342, 82344, 82357, 82359, 82375, 82394, 82397, 82403, 82430, 82438, 82448, 82477, 82479, 82481, 82482, 82532, 82534, 82542, 82603, 82606, 82634, 82639, 82652, 82673, 82675, 82676, 82684, 82686, 82687, 82690, 82751, 82934, 82940, 82987, 83111, 83112, 83128, 83129, 83132, 83169, 83176, 83177, 83190, 83193, 83194, 83195, 83197, 83200, 83201, 83202, 83203, 83204, 83206, 83208, 83210, 83236, 83240, 83242, 83243, 83246, 83247, 83249, 83250, 83251, 83252, 83253, 83257, 83258, 83275, 83277, 83278, 83290, 83291, 83293, 83294, 83300, 83306, 83309, 83311, 83314, 83315, 83316, 83320, 83322, 83335, 83336, 83337, 83339, 83343, 83354, 83355, 83363, 83368, 83381, 83383, 83477, 83478, 83479, 83480, 83543, 83548, 83580, 83594, 83633, 83635, 83636, 83638, 83639, 83642, 83643, 83647, 83653, 83654, 83656, 83657, 83659, 83660, 83666, 83696, 83840, 83852, 83860, 83925, 83927, 83928, 83929, 83930, 83932, 83933, 83935, 83936, 83937, 83939, 83941, 83942, 83943, 83954, 83957, 83961, 83966, 83967, 83971, 83972, 83973, 83974, 84028, 84032, 84034, 84039, 84041, 84042, 84062, 84063, 84069, 84070, 84076, 84078, 84080, 84099, 84100, 84102, 84103, 84104, 84105, 84106, 84107, 84110, 84114, 84115, 84117, 84118, 84119, 84120, 84121, 84122, 84123, 84132, 84133, 84137, 84138, 84139, 84141, 84142, 84145, 84146, 84156, 84164, 84165, 84310, 84320, 84321, 84322, 84327, 84333, 84334, 84335, 84336, 84337, 84338, 84339, 84345, 84346, 84347, 84349, 84356, 84357, 84358, 84359, 84362, 84363, 84368, 84369, 84370, 84373, 84376, 84385, 84391, 84392, 84396, 84399, 84455, 84459, 84523, 84537, 84551, 84553, 84559, 84562, 84563, 84565, 84572, 84583, 84591, 84597, 84598, 84611, 84612, 84620, 84651, 84665, 84686, 84689, 84690, 84691, 84736, 84739, 84753, 84754, 84755, 84757, 84758, 84759, 84764, 84765, 84785, 84793, 84797, 84798, 84805, 84809, 84810, 84824, 84826, 84827, 84828, 84829, 84830, 84832, 84833, 84835, 84836, 84837, 84838, 84839, 84841, 84842, 84865, 84866, 84867, 84868, 84870, 84872, 84874, 84875, 84877, 84881, 84883, 84899, 84900, 84901, 84981, 85000, 85015, 85022, 85023, 85024, 85027, 85028, 85029, 85031, 85039, 85040, 85041, 85042, 85043, 85044, 85045, 85048, 85049, 85050, 85051, 85054, 85074, 85088, 85110, 85111, 85191, 85192, 85194, 85195, 85196, 85197, 85200, 85201, 85262, 85273, 85275, 85277, 85278, 85279, 85281, 85282, 85283, 85284, 85288, 85289, 85291, 85292, 85299, 85303, 85304, 85306, 85308, 85310, 85334, 85347, 85352, 85380, 85456, 85458, 85459, 85462, 85471, 85579, 85580, 85582, 85645, 85646, 85650, 85651, 85653, 85654, 85656, 85658, 85685, 85686, 85687, 85689, 85691, 85692, 85693, 85694, 85696, 85697, 85698, 85699, 85701, 85714, 85715, 85718, 85719, 85720, 85722, 85723, 85724, 85725, 85726, 85728, 85729, 85730, 85765, 85828, 85862, 85903, 85945, 85946, 85947, 85948, 85950, 85952, 85969, 85972, 85974, 86044, 86045, 86077, 86227, 86229, 86232, 86247, 86303, 86304, 86306, 86308, 86309, 86311, 86312, 86314, 86315, 86317, 86320, 86321, 86322, 86323, 86325, 86327, 86330, 86340, 86341, 86342, 86343, 86344, 86346, 86349, 86351, 86352, 86366, 86368, 86369, 86370, 86371, 86372, 86373, 86375, 86377, 86378, 86380, 86384, 86398, 86406, 86410, 86415, 86416, 86417, 86420, 86425, 86426, 86428, 86429, 86433, 86439, 86440, 86492, 86510, 86511, 86523, 86544, 86547, 86551, 86553, 86557, 86559, 86567, 86594, 86620, 86667, 86675, 86677, 86678, 86681, 86682, 86683, 86684, 86688, 86691, 86692, 86696, 86700, 86703, 86704, 86707, 86708, 86715, 86717, 86720, 86721, 86722, 86723, 86724, 86725, 86727, 86729, 86730, 86731, 86733, 86734, 86736, 86738, 86739, 86741, 86743, 86744, 86748, 86755, 86758, 86760, 86762, 86844, 86981, 87038, 87050, 87052, 87053, 87055, 87056, 87057, 87059, 87060, 87062, 87064, 87065, 87068, 87069, 87070, 87074, 87077, 87078, 87079, 87081, 87084, 87086, 87087, 87088, 87090, 87092, 87101, 87103, 87107, 87108, 87109, 87111, 87112, 87113, 87117, 87118, 87120, 87122, 87124, 87125, 87126, 87127, 87128, 87129, 87131, 87133, 87137, 87139, 87140, 87142, 87143, 87144, 87145, 87148, 87149, 87150, 87168, 87171, 87218, 87220, 87223, 87225, 87236, 87261, 87300, 87320, 87325, 87342, 87344, 87347, 87367, 87369, 87382, 87406, 87407, 87408, 87410, 87499, 87500, 87503, 87510, 87523, 87524, 87526, 87527, 87541, 87544, 87547, 87549, 87550, 87552, 87569, 87571, 87575, 87576, 87577, 87584, 87585, 87587, 87589, 87597, 87599, 87606, 87607, 87863, 87865, 87870, 87871, 87875, 87881, 87910, 87911, 87992, 87994, 87995, 87996, 87997, 87998, 87999, 88000, 88001, 88006, 88008, 88010, 88012, 88014, 88015, 88019, 88020, 88021, 88023, 88026, 88033, 88035, 88036, 88038, 88059, 88070, 88072, 88091, 88092, 88093, 88094, 88096, 88097, 88098, 88099, 88100, 88102, 88105, 88112, 88126, 88127, 88130, 88131, 88132, 88133, 88134, 88135, 88137, 88138, 88140, 88141, 88142, 88144, 88146, 88150, 88151, 88152, 88153, 88154, 88156, 88157, 88158, 88166, 88168, 88169, 88171, 88172, 88173, 88174, 88175, 88176, 88177, 88178, 88183, 88186, 88187, 88188, 88192, 88193, 88198, 88199, 88200, 88201, 88203, 88206, 88208, 88212, 88213, 88218, 88219, 88220, 88221, 88231, 88232, 88316, 88318, 88319, 88320, 88415, 88430, 88476, 88479, 88481, 88483, 88484, 88516, 88517, 88518, 88591, 88638, 88639, 88640, 88642, 88644, 88645, 88649, 88652, 88653, 88656, 88688, 88744, 88765, 88777, 88786, 88787, 88789, 88790, 88793, 88795, 88797, 88799, 88800, 88803, 88817, 88820, 88821, 88824, 88826, 88827, 88829, 88831, 88832, 88908, 88910, 88911, 88913, 88915, 88916, 88917, 88918, 88971, 88972, 88973, 88975, 88989, 89016, 89018, 89019, 89048, 89049, 89050, 89051, 89052, 89053, 89054, 89055, 89082, 89085, 89087, 89088, 89089, 89091, 89092, 89093, 89094, 89095, 89096, 89097, 89099, 89100, 89102, 89103, 89104, 89106, 89107, 89109, 89110, 89111, 89112, 89113, 89114, 89115, 89116, 89117, 89119, 89120, 89124, 89125, 89153, 89154, 89155, 89156, 89157, 89158, 89159, 89160, 89161, 89162, 89163, 89165, 89166, 89167, 89168, 89169, 89170, 89171, 89175, 89176, 89190, 89192, 89193, 89194, 89196, 89197, 89199, 89203, 89204, 89205, 89206, 89208, 89209, 89211, 89213, 89215, 89222, 89249, 89251, 89252, 89338, 89339, 89341, 89343, 89344, 89345, 89347, 89348, 89354, 89355, 89359, 89360, 89361, 89362, 89364, 89365, 89369, 89370, 89371, 89372, 89375, 89377, 89378, 89418, 89442, 89443, 89444, 89448, 89450, 89451, 89452, 89453, 89454, 89485, 89541, 89548, 89550, 89551, 89552, 89554, 89569, 89675, 89878, 89884, 89929, 89930, 89932, 89933, 89937, 89938, 89939, 89941, 89942, 89943, 89946, 89949, 89952, 89955, 89959, 89960, 89980, 89983, 90040, 90133, 90134, 90144, 90146, 90160, 90162, 90179, 90182, 90184, 90185, 90187, 90190, 90192, 90194, 90205, 90208, 90433, 90504, 90505, 90506, 90511, 90512, 90513, 90514, 90515, 90516, 90517, 90518, 90519, 90520, 90522, 90540, 90547, 90648, 90650, 90651, 90654, 90696, 90698, 90704, 90705, 90715, 90813, 90815, 90817, 90820, 90822, 90823, 90828, 90833, 90896, 91000, 91079, 91115, 91120, 91132, 91158, 91159, 91186, 91187, 91188, 91234, 91238, 91266, 91268, 91309, 91322, 91338, 91353, 91357, 91397, 91410, 91412, 91414, 91438, 91441, 91445, 91448, 91462, 91469, 91470, 91471, 91473, 91474, 91571, 91574, 91576, 91615, 91621, 91642, 91753, 91754, 91755, 91756, 91757, 91759, 91760, 91762, 91784, 91786, 91788, 91793, 91797, 91798, 91801, 91804, 91807, 91808, 91811, 91812, 91817, 91818, 91820, 91821, 91822, 91838, 91839, 91846, 91847, 91848, 91850, 91854, 91858, 91866, 91867, 91869, 91871, 91872, 91874, 91892, 91897, 91898, 91899, 91901, 91903, 91904, 91905, 91907, 91908, 91909, 91910, 91911, 91913, 91914, 91915, 91920, 91921, 91922, 91923, 91924, 91926, 91927, 91930, 91932, 91935, 91944, 91952, 91956, 91961, 91962, 91963, 91965, 91966, 91967, 91968, 91969, 91970, 91972, 91973, 91975, 91976, 91977, 91980, 91982, 91984, 91985, 91987, 91988, 91989, 91992, 92005, 92015, 92032, 92038, 92103, 92121, 92128, 92136, 92171, 92172, 92173, 92174, 92208, 92232, 92233, 92234, 92236, 92238, 92245, 92246, 92247, 92248, 92250, 92251, 92355, 92356, 92358, 92389, 92440, 92492, 92495, 92502, 92503, 92506, 96211, 96212, 96214, 96215, 96216, 96217, 96218, 96219, 96220, 96221, 96224, 96225, 96226, 96227, 96228, 96229, 96230, 96231, 96234, 96235, 96236, 96238, 96240, 96242, 96243, 96244, 96245, 96246, 96248, 96249, 96255, 96258, 96259, 96261, 96262, 96264, 96265, 96266, 96267, 96268, 96269, 96270, 96271, 96272, 96274, 96276, 96277, 96279, 96280, 96281, 96285, 96287, 96289, 96291, 96292, 96294, 96295, 96296, 96297, 96299], "val_users": [70692, 70701, 70727, 70731, 70744, 70748, 70872, 71198, 71211, 71216, 71225, 71557, 71830, 72735, 73963, 75169, 77899, 77949, 77954, 77970, 77994, 78003, 78020, 78023, 78073, 78107, 78155, 78158, 78164, 78171, 78358, 78373, 78391, 78446, 78458, 78528, 78544, 78555, 78566, 78572, 78575, 78577, 78723, 78726, 78732, 78777, 78785, 78834, 78874, 78904, 78915, 78921, 78922, 78970, 78992, 78993, 79007, 79019, 79030, 79069, 79075, 79214, 79236, 79279, 79427, 79496, 79499, 79507, 79515, 79541, 79569, 79579, 79683, 79698, 79785, 79862, 80013, 80021, 80024, 80091, 80110, 80111, 80116, 80176, 80177, 80179, 80181, 80216, 80217, 80242, 80610, 80655, 80719, 80916, 80924, 80992, 81196, 81470, 81643, 81892, 82094, 82273, 82329, 82341, 82391, 82476, 82616, 82681, 83199, 83292, 83302, 83312, 83578, 83630, 83640, 83641, 83662, 83876, 83940, 84044, 84169, 84178, 84289, 84314, 84316, 84341, 84374, 84518, 84554, 84564, 84667, 84749, 84750, 84752, 84760, 84821, 84895, 84902, 84999, 85007, 85037, 85133, 85276, 85297, 85300, 85551, 85949, 85951, 86248, 86318, 86338, 86345, 86348, 86374, 86382, 86407, 86424, 86552, 86556, 86631, 86666, 86701, 86713, 86726, 86735, 86742, 86749, 86751, 86754, 86763, 87051, 87054, 87063, 87116, 87216, 87340, 87409, 87421, 87468, 87525, 87866, 87990, 87993, 88034, 88069, 88128, 88147, 88148, 88170, 88180, 88214, 88224, 88693, 88830, 88833, 89046, 89083, 89098, 89173, 89198, 89202, 89247, 89356, 89928, 90167, 90188, 90523, 90538, 90707, 90819, 90924, 91408, 91443, 91467, 91763, 91802, 91810, 91843, 91865, 91925, 91928, 91933, 91934, 91983, 91991, 92072, 92137, 92235, 92301, 92441, 92521, 96213, 96222, 96223, 96247, 96250, 96253, 96254, 96256, 96283]}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN128-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN128-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..fc609de9d645e7f5f5d795d55ec0fd1d70a3cb35
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN128-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 33.6, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.686676006087919, "n_train_users": 102, "n_train_chunks": 526, "n_dev_chunks": 126, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 42.2, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN128-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN128-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..fcc2a60add58ce2283506b20e2878475dd8f7b62
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN128-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 29.1, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.676787100285477, "n_train_users": 102, "n_train_chunks": 475, "n_dev_chunks": 102, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 37.2, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN128-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN128-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..35ea23b8a228c78f0f33f3fc7572d8af79eb1260
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN128-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 31.9, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.6192180117876712, "n_train_users": 102, "n_train_chunks": 492, "n_dev_chunks": 139, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 46.6, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN128-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN128-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e704774984e2cd4608ee8680ce10f8ca66b7c3dd
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN128-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 32.4, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.6511667763351398, "n_train_users": 102, "n_train_chunks": 451, "n_dev_chunks": 126, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 43.7, "flip_max_diff": 0.0, "peak_gb": 9.12}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN128-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN128-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..768c86f87cae7f3bb2ec12b09341a43f18dddba4
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN128-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 25.3, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6268495610494038, "n_train_users": 102, "n_train_chunks": 397, "n_dev_chunks": 92, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 42.5, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN16-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN16-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..80faeba3dd34e6d74be49f1ba615b9a6a70377d1
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN16-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 12.0, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.6263143851318606, "n_train_users": 13, "n_train_chunks": 56, "n_dev_chunks": 13, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 80.3, "flip_max_diff": 0.0, "peak_gb": 7.99}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN16-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN16-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..680e32fa008675979c6d059e39ac024d50d81164
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN16-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 7.9, "epochs": 17, "best_epoch": 7, "dev_auc_pykt": 0.6585554626791741, "n_train_users": 13, "n_train_chunks": 54, "n_dev_chunks": 13, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 36.3, "flip_max_diff": 0.0, "peak_gb": 7.7}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN16-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN16-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..c7bf94a8ef60c5ec72de50f2d91dd81b49bab03f
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN16-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 11.3, "epochs": 26, "best_epoch": 16, "dev_auc_pykt": 0.6242735936773594, "n_train_users": 13, "n_train_chunks": 64, "n_dev_chunks": 14, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 42.8, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN16-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN16-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..c2780ace72d3e642aba26ad74f623c6a35bbf542
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN16-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 5.9, "epochs": 16, "best_epoch": 6, "dev_auc_pykt": 0.5725959490337322, "n_train_users": 13, "n_train_chunks": 36, "n_dev_chunks": 13, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 34.1, "flip_max_diff": 0.0, "peak_gb": 5.16}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN16-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN16-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..92d1a7c4eb1e3cc2045be57da5b6617a62b9a46f
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN16-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4.0, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.5372422366963054, "n_train_users": 13, "n_train_chunks": 53, "n_dev_chunks": 9, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 39.2, "flip_max_diff": 0.0, "peak_gb": 7.56}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN256-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN256-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6a6769b47aa54a165698d14cb90e61bf92c82dd2
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN256-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 59.5, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6397336290312657, "n_train_users": 205, "n_train_chunks": 1017, "n_dev_chunks": 216, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 48.3, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN256-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN256-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b256039ae97c96ff852809d79bd9916325d18f5a
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN256-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 62.5, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6408422168256671, "n_train_users": 205, "n_train_chunks": 954, "n_dev_chunks": 240, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 48.4, "flip_max_diff": 0.0, "peak_gb": 9.12}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN256-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN256-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..f2d3bfeace64c66ec9b07dc63c83c8f861560f84
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN256-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 58.6, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6241910446095956, "n_train_users": 205, "n_train_chunks": 955, "n_dev_chunks": 227, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 48.0, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN256-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN256-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..0d581da5b8de0df137f21fdf7e95982161959f28
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN256-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 56.6, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6480666505478857, "n_train_users": 205, "n_train_chunks": 910, "n_dev_chunks": 225, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 48.3, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN256-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN256-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..953118171fd9eab062e98b450556789090e767d9
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN256-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 54.1, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6523345910151928, "n_train_users": 205, "n_train_chunks": 838, "n_dev_chunks": 209, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 49.8, "flip_max_diff": 0.0, "peak_gb": 9.12}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN32-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN32-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b1b6b9272752a00fa9a89a4397704195df8f3949
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN32-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.4, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.6417292275722973, "n_train_users": 26, "n_train_chunks": 103, "n_dev_chunks": 43, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 44.8, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN32-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN32-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..38ed322f60ee31ffe2e0b005ec78199aee266d44
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN32-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 13.0, "epochs": 17, "best_epoch": 7, "dev_auc_pykt": 0.6383129542701392, "n_train_users": 26, "n_train_chunks": 117, "n_dev_chunks": 30, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 35.1, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN32-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN32-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6e0021c4ad8b0505d94d72ada3a7582a990da33c
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN32-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 11.6, "epochs": 16, "best_epoch": 6, "dev_auc_pykt": 0.6191965231538603, "n_train_users": 26, "n_train_chunks": 134, "n_dev_chunks": 28, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 43.3, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN32-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN32-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..862389c9dba5f6256ebe8111ee90aceac416d5ad
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN32-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 12.1, "epochs": 16, "best_epoch": 6, "dev_auc_pykt": 0.6451543466604339, "n_train_users": 26, "n_train_chunks": 91, "n_dev_chunks": 37, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 40.5, "flip_max_diff": 0.0, "peak_gb": 9.12}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN32-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN32-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..cbcd6c8ad2b09da2941b92da2a0e1851fdd10782
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN32-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.0, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.606325111696917, "n_train_users": 26, "n_train_chunks": 118, "n_dev_chunks": 15, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 24.4, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN64-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN64-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a566640440afca818d4afaca8fb6fb2fd1296e9f
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN64-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 16.4, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.6496466435083301, "n_train_users": 51, "n_train_chunks": 263, "n_dev_chunks": 71, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 43.1, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN64-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN64-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..daab3c0a976273031496875a6a401d2ac17d284f
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN64-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 15.7, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.6396053063694198, "n_train_users": 51, "n_train_chunks": 249, "n_dev_chunks": 80, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 45.9, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN64-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN64-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..86c1bd0b9853256a0055341cf9d72209084e2a26
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN64-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 17.0, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.6076376403678614, "n_train_users": 51, "n_train_chunks": 250, "n_dev_chunks": 54, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 43.5, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN64-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN64-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8a37e353d8c230c31deb2f24884045c736a677a9
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN64-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 16.2, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.6723337171962115, "n_train_users": 51, "n_train_chunks": 199, "n_dev_chunks": 48, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 45.2, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN64-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN64-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b24948022d656cdc20e5e2638518b4532c862fad
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN64-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.3, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.6842407106829327, "n_train_users": 51, "n_train_chunks": 199, "n_dev_chunks": 57, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 25.9, "flip_max_diff": 0.0, "peak_gb": 9.12}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN8-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN8-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..be32f50de6154ad0fab09d1d5991585ae143f49c
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN8-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 15.8, "epochs": 27, "best_epoch": 17, "dev_auc_pykt": 0.6118455370582887, "n_train_users": 6, "n_train_chunks": 28, "n_dev_chunks": 10, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 73.4, "flip_max_diff": 0.0, "peak_gb": 4.04}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN8-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN8-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..f7e6c56973c80d05080fdfb9ea87f9c99e4207a8
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN8-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4.2, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.5163239538239539, "n_train_users": 6, "n_train_chunks": 30, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 34.6, "flip_max_diff": 0.0, "peak_gb": 4.32}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN8-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN8-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8fe6770f9e0464ed7722ccd56a57dd852b2a3dbb
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN8-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.2, "epochs": 24, "best_epoch": 14, "dev_auc_pykt": 0.6247123384669854, "n_train_users": 6, "n_train_chunks": 35, "n_dev_chunks": 7, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 41.7, "flip_max_diff": 0.0, "peak_gb": 5.02}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN8-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN8-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..9efb46c69e5beae4dff6b8bb2f7fd94d0d4ec86f
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN8-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 7.2, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.6026107133550893, "n_train_users": 6, "n_train_chunks": 22, "n_dev_chunks": 7, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 38.3, "flip_max_diff": 0.0, "peak_gb": 3.2}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csN8-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csN8-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..34d6ec68b144f69f35e53f387727a6bad4e0b618
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csN8-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.3, "epochs": 17, "best_epoch": 7, "dev_auc_pykt": 0.5519667891656459, "n_train_users": 6, "n_train_chunks": 16, "n_dev_chunks": 7, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 23.7, "flip_max_diff": 0.0, "peak_gb": 2.34}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csNall-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csNall-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b35b9bedea940f299a09532afcff2982ec0e8c36
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csNall-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3006.3, "epochs": 108, "best_epoch": 98, "dev_auc_pykt": 0.7590204680104826, "n_train_users": 983, "n_train_chunks": 4659, "n_dev_chunks": 1096, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 50.3, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csNall-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csNall-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..9a9ea6df36ef2779d69979608c1bff8c6922547c
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csNall-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2847.4, "epochs": 102, "best_epoch": 92, "dev_auc_pykt": 0.7676099355893012, "n_train_users": 983, "n_train_chunks": 4602, "n_dev_chunks": 1153, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 50.4, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csNall-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csNall-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..89ed3381ef3d1eda68e88233cc618c9a912428f2
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csNall-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2774.7, "epochs": 97, "best_epoch": 87, "dev_auc_pykt": 0.7664143543839915, "n_train_users": 983, "n_train_chunks": 4611, "n_dev_chunks": 1144, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 52.4, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csNall-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csNall-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..78e9324a0745f0606b5babf28679b9efc036e915
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csNall-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2234.3, "epochs": 81, "best_epoch": 71, "dev_auc_pykt": 0.7621542330832074, "n_train_users": 983, "n_train_chunks": 4576, "n_dev_chunks": 1179, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 39.0, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-csNall-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-csNall-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..eed94ae4136393f530eaf7e93b18eb5bfdb31acf
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-csNall-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2677.6, "epochs": 94, "best_epoch": 84, "dev_auc_pykt": 0.7674435933415128, "n_train_users": 983, "n_train_chunks": 4613, "n_dev_chunks": 1142, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 48.5, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-ics0/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-ics0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..955bcf25f174ca4ba6f170ea16c43d0ea1798642
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-ics0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2731.1, "epochs": 107, "best_epoch": 97, "dev_auc_pykt": 0.753335858326416, "n_train_users": 983, "n_train_chunks": 3812, "n_dev_chunks": 901, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 3.5, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-akt-ics1/train_info.json.done b/pykt/assist2017/coldstart/pykt-akt-ics1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e61ebc674acdf26c2610ee6afd38a5cbf5b04d9a
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-akt-ics1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2383.8, "epochs": 105, "best_epoch": 95, "dev_auc_pykt": 0.7658328254255073, "n_train_users": 983, "n_train_chunks": 3759, "n_dev_chunks": 950, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 1.9, "flip_max_diff": 0.0, "peak_gb": 9.11}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN128-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN128-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3aa2d68fed26397d704347b459c6a4ebd120718b
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN128-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.4, "epochs": 54, "best_epoch": 44, "dev_auc_pykt": 0.6831687067544515, "n_train_users": 102, "n_train_chunks": 526, "n_dev_chunks": 126, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 3.6, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN128-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN128-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..370d1daa94e5c400c61b9f3bbef26f3e3e59b7d5
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN128-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4.1, "epochs": 19, "best_epoch": 9, "dev_auc_pykt": 0.622916656906105, "n_train_users": 102, "n_train_chunks": 475, "n_dev_chunks": 102, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 5.2, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN128-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN128-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2004260cce9a95c9af75ab562d71f0eb09e21fab
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN128-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 7.4, "epochs": 43, "best_epoch": 33, "dev_auc_pykt": 0.6553837984178541, "n_train_users": 102, "n_train_chunks": 492, "n_dev_chunks": 139, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 4.9, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN128-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN128-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..bbd2f2ce6db7a7d11151516a7e127dc0afeaad1b
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN128-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 10.4, "epochs": 45, "best_epoch": 35, "dev_auc_pykt": 0.6815727847592604, "n_train_users": 102, "n_train_chunks": 451, "n_dev_chunks": 126, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 4.5, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN128-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN128-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..78c361f43114fe3954208598b18771f1653c9f8e
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN128-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.8, "epochs": 44, "best_epoch": 34, "dev_auc_pykt": 0.6387388045171339, "n_train_users": 102, "n_train_chunks": 397, "n_dev_chunks": 92, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 4.7, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN16-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN16-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..64b1441474400531167a6ec471d7f27ab2d83f2c
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN16-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.1, "epochs": 36, "best_epoch": 26, "dev_auc_pykt": 0.6143943890525536, "n_train_users": 13, "n_train_chunks": 56, "n_dev_chunks": 13, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 4.2, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN16-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN16-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..f868c1e3c2887f417b439abafc68a88cfab22a24
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN16-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4.9, "epochs": 24, "best_epoch": 14, "dev_auc_pykt": 0.6118270332187858, "n_train_users": 13, "n_train_chunks": 54, "n_dev_chunks": 13, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 4.1, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN16-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN16-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..d392deeb65b10f5dbfeb1f84e5acc6a6dc19e847
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN16-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3.1, "epochs": 48, "best_epoch": 38, "dev_auc_pykt": 0.6186463660313032, "n_train_users": 13, "n_train_chunks": 64, "n_dev_chunks": 14, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 4.1, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN16-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN16-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..c546b9d8293d885dbb5d004f877eef699e452f72
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN16-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.9, "epochs": 19, "best_epoch": 9, "dev_auc_pykt": 0.5799321703818042, "n_train_users": 13, "n_train_chunks": 36, "n_dev_chunks": 13, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 3.8, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN16-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN16-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8e2e5bbda06e88cb0cd29ac580c82da1289294f0
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN16-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.4, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.5517891408340315, "n_train_users": 13, "n_train_chunks": 53, "n_dev_chunks": 9, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 3.5, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN256-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN256-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1cc0ee3a4025d3d51232316b70a048f1b2996387
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN256-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 12.1, "epochs": 39, "best_epoch": 29, "dev_auc_pykt": 0.692401344239152, "n_train_users": 205, "n_train_chunks": 1017, "n_dev_chunks": 216, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 3.6, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN256-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN256-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ac20f1ad184768c998ae7a29d2a10ae8cc0ca770
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN256-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 15.0, "epochs": 41, "best_epoch": 31, "dev_auc_pykt": 0.6825832616478704, "n_train_users": 205, "n_train_chunks": 954, "n_dev_chunks": 240, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 5.5, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN256-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN256-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2d7366e70b6f45d0c84d4b470ba3d8e644bb0aab
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN256-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 14.7, "epochs": 32, "best_epoch": 22, "dev_auc_pykt": 0.6678788366317149, "n_train_users": 205, "n_train_chunks": 955, "n_dev_chunks": 227, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 4.7, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN256-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN256-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3eb00a12c604a4cc611a640b09c7409f18040c29
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN256-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 13.7, "epochs": 35, "best_epoch": 25, "dev_auc_pykt": 0.685476419944554, "n_train_users": 205, "n_train_chunks": 910, "n_dev_chunks": 225, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 4.8, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN256-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN256-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2089b5d2eccccac661152772ceed0c26ddcb9ec7
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN256-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.7, "epochs": 35, "best_epoch": 25, "dev_auc_pykt": 0.6849761726974345, "n_train_users": 205, "n_train_chunks": 838, "n_dev_chunks": 209, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 3.8, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN32-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN32-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7d973e33b855624cb5e68cecab67d5b24904cd42
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN32-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.8, "epochs": 23, "best_epoch": 13, "dev_auc_pykt": 0.6043230689307433, "n_train_users": 26, "n_train_chunks": 103, "n_dev_chunks": 43, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 3.7, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN32-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN32-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6f45d953d4458ddde111d432702377a6137ab79e
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN32-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4.9, "epochs": 26, "best_epoch": 16, "dev_auc_pykt": 0.6045852892186554, "n_train_users": 26, "n_train_chunks": 117, "n_dev_chunks": 30, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 5.1, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN32-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN32-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7fdb8dcfa203e39117ffe50aba275116e1d99f21
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN32-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.9, "epochs": 25, "best_epoch": 15, "dev_auc_pykt": 0.594010121918474, "n_train_users": 26, "n_train_chunks": 134, "n_dev_chunks": 28, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 4.1, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN32-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN32-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1b800048fcb54f62724569c9caed69d3b25b204a
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN32-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3.0, "epochs": 24, "best_epoch": 14, "dev_auc_pykt": 0.5854474867324773, "n_train_users": 26, "n_train_chunks": 91, "n_dev_chunks": 37, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 4.7, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN32-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN32-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7d2abc7780c44f2f27305ad819e008bc64b7de9b
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN32-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.0, "epochs": 16, "best_epoch": 6, "dev_auc_pykt": 0.5929095354523228, "n_train_users": 26, "n_train_chunks": 118, "n_dev_chunks": 15, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 3.7, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN64-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN64-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6d5c6ecfddb3ada50129e9c59629a369759c986d
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN64-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.3, "epochs": 21, "best_epoch": 11, "dev_auc_pykt": 0.613520541173762, "n_train_users": 51, "n_train_chunks": 263, "n_dev_chunks": 71, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 3.6, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN64-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN64-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1c927ed027ad8481bd558f10209d38518234a0cf
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN64-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 5.5, "epochs": 29, "best_epoch": 19, "dev_auc_pykt": 0.6201681572959241, "n_train_users": 51, "n_train_chunks": 249, "n_dev_chunks": 80, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 4.4, "flip_max_diff": 0.0, "peak_gb": 1.04}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN64-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN64-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..fc650128702b4608cf1c2fca144e69b3df2eb964
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN64-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3.2, "epochs": 28, "best_epoch": 18, "dev_auc_pykt": 0.5882378565913278, "n_train_users": 51, "n_train_chunks": 250, "n_dev_chunks": 54, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 7.4, "flip_max_diff": 0.0, "peak_gb": 1.04}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN64-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN64-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..87546369cde4439b8ac9e8b7755ff9a341e84d2b
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN64-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.8, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.6336969461763368, "n_train_users": 51, "n_train_chunks": 199, "n_dev_chunks": 48, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 3.9, "flip_max_diff": 0.0, "peak_gb": 0.83}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN64-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN64-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3e263df58eefeb7e973d7b91c3ea18f310286631
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN64-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3.0, "epochs": 23, "best_epoch": 13, "dev_auc_pykt": 0.6381431837618026, "n_train_users": 51, "n_train_chunks": 199, "n_dev_chunks": 57, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 4.5, "flip_max_diff": 0.0, "peak_gb": 0.83}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN8-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN8-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..4f621e3931b4f65e856a4c52d00c6a2810826609
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN8-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.3, "epochs": 39, "best_epoch": 29, "dev_auc_pykt": 0.5742134521662667, "n_train_users": 6, "n_train_chunks": 28, "n_dev_chunks": 10, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 3.8, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN8-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN8-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..dade08a2ce61f591b6d8275aacc088cd85852fa4
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN8-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.2, "epochs": 34, "best_epoch": 24, "dev_auc_pykt": 0.4995490620490621, "n_train_users": 6, "n_train_chunks": 30, "n_dev_chunks": 6, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 4.5, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN8-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN8-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..c4b13bb7c509f836822f8f666ea2bccaaf3357a1
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN8-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.2, "epochs": 24, "best_epoch": 14, "dev_auc_pykt": 0.49854588675618944, "n_train_users": 6, "n_train_chunks": 35, "n_dev_chunks": 7, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 4.3, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN8-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN8-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..29dafd6ab63e146a6b9300d7455140146f976007
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN8-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.1, "epochs": 26, "best_epoch": 16, "dev_auc_pykt": 0.5508927795907732, "n_train_users": 6, "n_train_chunks": 22, "n_dev_chunks": 7, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 3.9, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csN8-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csN8-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6a7aad203c2002f6acf2fc0b423a4d54f15fbc33
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csN8-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.2, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.5357152579284061, "n_train_users": 6, "n_train_chunks": 16, "n_dev_chunks": 7, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 4.2, "flip_max_diff": 0.0, "peak_gb": 0.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csNall-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csNall-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..fac2a9affad4a06ff9539dc472e54f39cfc83732
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csNall-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 42.2, "epochs": 25, "best_epoch": 15, "dev_auc_pykt": 0.7190416096477442, "n_train_users": 983, "n_train_chunks": 4659, "n_dev_chunks": 1096, "batch_size": 256, "seq_len": 200, "n_val_users": 246, "score_s": 4.1, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csNall-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csNall-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..31cbf95bffc4f6a9522e1324c916f36506c9f1d7
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csNall-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 54.3, "epochs": 24, "best_epoch": 14, "dev_auc_pykt": 0.7355065230590349, "n_train_users": 983, "n_train_chunks": 4602, "n_dev_chunks": 1153, "batch_size": 256, "seq_len": 200, "n_val_users": 246, "score_s": 5.5, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csNall-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csNall-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..f3120a7fc5fe3ea8dc167d5b7e7a1bdc5a56a70c
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csNall-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 55.2, "epochs": 25, "best_epoch": 15, "dev_auc_pykt": 0.7290070513228141, "n_train_users": 983, "n_train_chunks": 4611, "n_dev_chunks": 1144, "batch_size": 256, "seq_len": 200, "n_val_users": 246, "score_s": 5.4, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csNall-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csNall-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a894ca3a89636b306ad790e53fdb218f9540afa1
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csNall-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 49.0, "epochs": 24, "best_epoch": 14, "dev_auc_pykt": 0.7289451687074665, "n_train_users": 983, "n_train_chunks": 4576, "n_dev_chunks": 1179, "batch_size": 256, "seq_len": 200, "n_val_users": 246, "score_s": 4.9, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-csNall-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-csNall-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ee6f7f30db5d12f589c1ed38ea24108546784748
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-csNall-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 51.8, "epochs": 26, "best_epoch": 16, "dev_auc_pykt": 0.7208351240185913, "n_train_users": 983, "n_train_chunks": 4613, "n_dev_chunks": 1142, "batch_size": 256, "seq_len": 200, "n_val_users": 246, "score_s": 4.8, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atdkt-ics0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atdkt-ics0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3217db8ed06cd16a65c493c920cc34e76410d7e3
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atdkt-ics0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 39.9, "epochs": 23, "best_epoch": 13, "dev_auc_pykt": 0.713684268466959, "n_train_users": 983, "n_train_chunks": 3812, "n_dev_chunks": 901, "batch_size": 256, "seq_len": 200, "n_val_users": 246, "score_s": 0.3, "flip_max_diff": 0.0, "peak_gb": 1.07}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN128-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN128-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7f613f19cff5088c765b4346a1f03bcf6cdb3e95
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN128-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 60.9, "epochs": 80, "best_epoch": 70, "dev_auc_pykt": 0.6402331246525981, "n_train_users": 102, "n_train_chunks": 526, "n_dev_chunks": 126, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 2.9, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN128-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN128-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b2c834bdd1c17f2c6ea0198076e64d626378317f
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN128-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 59.6, "epochs": 91, "best_epoch": 81, "dev_auc_pykt": 0.6442017090353129, "n_train_users": 102, "n_train_chunks": 475, "n_dev_chunks": 102, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 4.3, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN128-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN128-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a4bfd5dba217aa9e208d196a9c0ef61357ae46a0
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN128-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 174.7, "epochs": 127, "best_epoch": 117, "dev_auc_pykt": 0.5937866055513115, "n_train_users": 102, "n_train_chunks": 492, "n_dev_chunks": 139, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 2.9, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN128-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN128-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..0047f02d35bcb279fe4f0795a42fc8d94786f2f3
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN128-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 41.9, "epochs": 118, "best_epoch": 108, "dev_auc_pykt": 0.6297662761324514, "n_train_users": 102, "n_train_chunks": 451, "n_dev_chunks": 126, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 4.8, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN128-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN128-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..19c7b78843b4ed20762a49557df9a71ad5ec0abd
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN128-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 29.3, "epochs": 74, "best_epoch": 64, "dev_auc_pykt": 0.5976166210998026, "n_train_users": 102, "n_train_chunks": 397, "n_dev_chunks": 92, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 3.6, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN16-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN16-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ecbe7a21c2ddc775922ed2aa13b492578386be0c
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN16-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.9, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.5085247847702855, "n_train_users": 13, "n_train_chunks": 56, "n_dev_chunks": 13, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 4.0, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN16-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN16-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b8874289f0674c2d07884bb1e05b9441452a309a
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN16-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.9, "epochs": 26, "best_epoch": 16, "dev_auc_pykt": 0.577888455465775, "n_train_users": 13, "n_train_chunks": 54, "n_dev_chunks": 13, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 3.5, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN16-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN16-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..79d498f578f902d4c224720be9f18313f723553d
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN16-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.2, "epochs": 43, "best_epoch": 33, "dev_auc_pykt": 0.5493859445219278, "n_train_users": 13, "n_train_chunks": 64, "n_dev_chunks": 14, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 4.7, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN16-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN16-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b3d16defbff9f6b6ec7a42aa9b8ee70c1ee89ae0
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN16-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.5, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.49323363596629544, "n_train_users": 13, "n_train_chunks": 36, "n_dev_chunks": 13, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 3.1, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN16-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN16-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6c8144214790185f2438d58e2dba34117ee91d63
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN16-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.9, "epochs": 19, "best_epoch": 9, "dev_auc_pykt": 0.5448636523337207, "n_train_users": 13, "n_train_chunks": 53, "n_dev_chunks": 9, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 2.8, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN256-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN256-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8eecc929e4249f066873d0da591ba8130f69992e
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN256-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 42.6, "epochs": 56, "best_epoch": 46, "dev_auc_pykt": 0.6324742038522998, "n_train_users": 205, "n_train_chunks": 1017, "n_dev_chunks": 216, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 2.8, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN256-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN256-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..95f5a3d044b9e26df145f30267a96d7e30af8d70
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN256-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 56.9, "epochs": 56, "best_epoch": 46, "dev_auc_pykt": 0.6262867696160865, "n_train_users": 205, "n_train_chunks": 954, "n_dev_chunks": 240, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 3.9, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN256-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN256-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..925a40abc9c92711279dfaffae0f23df27c826de
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN256-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 62.1, "epochs": 63, "best_epoch": 53, "dev_auc_pykt": 0.6087405917860793, "n_train_users": 205, "n_train_chunks": 955, "n_dev_chunks": 227, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 3.8, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN256-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN256-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2915abb50390ec81e12efbb617ef6e5318557a5a
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN256-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 51.7, "epochs": 57, "best_epoch": 47, "dev_auc_pykt": 0.6314583411915766, "n_train_users": 205, "n_train_chunks": 910, "n_dev_chunks": 225, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 3.7, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN256-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN256-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..24d8e096e43b4de678b005d0384d3635838be039
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN256-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 39.8, "epochs": 52, "best_epoch": 42, "dev_auc_pykt": 0.6563227453630698, "n_train_users": 205, "n_train_chunks": 838, "n_dev_chunks": 209, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 3.3, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN32-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN32-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2d9449ce14aa7e18349101e363a59ae1d376297a
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN32-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.0, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.5259944382760693, "n_train_users": 26, "n_train_chunks": 103, "n_dev_chunks": 43, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 2.8, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN32-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN32-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2339eb6a623f11d6a16e00cd21b0b8b7e9b928dc
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN32-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 19.1, "epochs": 102, "best_epoch": 92, "dev_auc_pykt": 0.5867021880678377, "n_train_users": 26, "n_train_chunks": 117, "n_dev_chunks": 30, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 4.7, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN32-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN32-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..9ae8d6a5132eb49cfae5e0600deb61d5d35055c5
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN32-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.2, "epochs": 17, "best_epoch": 7, "dev_auc_pykt": 0.522491641543145, "n_train_users": 26, "n_train_chunks": 134, "n_dev_chunks": 28, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 3.3, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN32-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN32-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..f64250abb4795b3e0bfc297f6c9a2a36e44ec473
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN32-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.7, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.5332384098220915, "n_train_users": 26, "n_train_chunks": 91, "n_dev_chunks": 37, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 3.6, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN32-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN32-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..dfcb5ad306a3d4ce237d3205c1557c4ce7149b7e
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN32-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.7, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.5569514904868472, "n_train_users": 26, "n_train_chunks": 118, "n_dev_chunks": 15, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 2.8, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN64-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN64-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..14c5a336623266e117e80a823227d398ec55fbcf
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN64-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 19.2, "epochs": 84, "best_epoch": 74, "dev_auc_pykt": 0.5994860474407329, "n_train_users": 51, "n_train_chunks": 263, "n_dev_chunks": 71, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 2.9, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN64-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN64-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..64cdfa39ea020a05ae865ae5e589236814eb2779
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN64-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.9, "epochs": 26, "best_epoch": 16, "dev_auc_pykt": 0.5669589883636901, "n_train_users": 51, "n_train_chunks": 249, "n_dev_chunks": 80, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 3.7, "flip_max_diff": 0.0, "peak_gb": 1.42}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN64-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN64-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1ddf13b38b690012916919afb07ce6ff06de2302
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN64-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 55.2, "epochs": 33, "best_epoch": 23, "dev_auc_pykt": 0.5423997990457066, "n_train_users": 51, "n_train_chunks": 250, "n_dev_chunks": 54, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 5.8, "flip_max_diff": 0.0, "peak_gb": 1.43}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN64-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN64-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..843af60eb53849e7e011f513cf2f0db9d7cf6c65
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN64-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 14.4, "epochs": 80, "best_epoch": 70, "dev_auc_pykt": 0.6007654851468741, "n_train_users": 51, "n_train_chunks": 199, "n_dev_chunks": 48, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 3.3, "flip_max_diff": 0.0, "peak_gb": 1.14}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN64-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN64-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3be44c31cb3aaca958af72eedf0488b65a343157
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN64-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.8, "epochs": 31, "best_epoch": 21, "dev_auc_pykt": 0.5492495244079096, "n_train_users": 51, "n_train_chunks": 199, "n_dev_chunks": 57, "batch_size": 256, "seq_len": 200, "n_val_users": 13, "score_s": 3.7, "flip_max_diff": 0.0, "peak_gb": 1.14}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN8-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN8-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..c1f2239bc205d90ac075fbcb9750bc34639f60fe
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN8-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.5, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.560934169933082, "n_train_users": 6, "n_train_chunks": 28, "n_dev_chunks": 10, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 2.9, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN8-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN8-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8c47206f32b15cc1bee203b6e263fb6d4c9d9261
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN8-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.2, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.4885461760461761, "n_train_users": 6, "n_train_chunks": 30, "n_dev_chunks": 6, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 3.2, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN8-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN8-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..5a59ba335a1e0ffe36c0b520342a23271fa75905
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN8-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.5, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.5597956654780871, "n_train_users": 6, "n_train_chunks": 35, "n_dev_chunks": 7, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 3.5, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN8-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN8-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..0a258f0afaa70ba26826d316845ff6a6ef1cb1e4
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN8-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 0.6, "epochs": 16, "best_epoch": 6, "dev_auc_pykt": 0.5314985544291546, "n_train_users": 6, "n_train_chunks": 22, "n_dev_chunks": 7, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 3.0, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csN8-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csN8-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3912fe8ce53da1aaf8c19c435b3096ad23c1f0d5
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csN8-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 1.0, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.5192595617258745, "n_train_users": 6, "n_train_chunks": 16, "n_dev_chunks": 7, "batch_size": 256, "seq_len": 200, "n_val_users": 2, "score_s": 3.4, "flip_max_diff": 0.0, "peak_gb": 0.8}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csNall-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csNall-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..5484be4e5c717ad241e1cd66b80794c3d7d8774b
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csNall-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 83.1, "epochs": 21, "best_epoch": 11, "dev_auc_pykt": 0.6215469431537993, "n_train_users": 983, "n_train_chunks": 4659, "n_dev_chunks": 1096, "batch_size": 256, "seq_len": 200, "n_val_users": 246, "score_s": 3.1, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csNall-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csNall-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1718e3bdf4686d7ccd6b1b11fdba2108d72e7bd6
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csNall-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 96.3, "epochs": 20, "best_epoch": 10, "dev_auc_pykt": 0.6412454487705832, "n_train_users": 983, "n_train_chunks": 4602, "n_dev_chunks": 1153, "batch_size": 256, "seq_len": 200, "n_val_users": 246, "score_s": 3.9, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csNall-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csNall-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6f00089957d5b8adf514d84ff2a71ecb5eec8980
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csNall-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 105.7, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.6463531442585624, "n_train_users": 983, "n_train_chunks": 4611, "n_dev_chunks": 1144, "batch_size": 256, "seq_len": 200, "n_val_users": 246, "score_s": 3.9, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csNall-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csNall-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..cb1b5c50dc4c08979c3176a01c8335f5b8aa6538
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csNall-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 97.8, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.63954564704512, "n_train_users": 983, "n_train_chunks": 4576, "n_dev_chunks": 1179, "batch_size": 256, "seq_len": 200, "n_val_users": 246, "score_s": 4.0, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-csNall-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-csNall-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ec55867675a87b9221f0205f444b10cffa1efc98
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-csNall-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 80.7, "epochs": 19, "best_epoch": 9, "dev_auc_pykt": 0.630185358681179, "n_train_users": 983, "n_train_chunks": 4613, "n_dev_chunks": 1142, "batch_size": 256, "seq_len": 200, "n_val_users": 246, "score_s": 3.5, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-atktfix-ics0/train_info.json.done b/pykt/assist2017/coldstart/pykt-atktfix-ics0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ba9d9e416af3a715b0c4dc2557c1b1ebd458a283
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-atktfix-ics0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 109.6, "epochs": 23, "best_epoch": 13, "dev_auc_pykt": 0.6246586288819689, "n_train_users": 983, "n_train_chunks": 3812, "n_dev_chunks": 901, "batch_size": 256, "seq_len": 200, "n_val_users": 246, "score_s": 0.2, "flip_max_diff": 0.0, "peak_gb": 1.46}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-deep_irt-csNall-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-deep_irt-csNall-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..f134499cb0f916442805798e57dbf8662bac955f
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-deep_irt-csNall-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 933.5, "epochs": 84, "best_epoch": 74, "dev_auc_pykt": 0.7108555289414665, "n_train_users": 983, "n_train_chunks": 4613, "n_dev_chunks": 1142, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 22.0, "flip_max_diff": 0.0, "peak_gb": 2.64}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-deep_irt-ics0/train_info.json.done b/pykt/assist2017/coldstart/pykt-deep_irt-ics0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..580da28caf668f5baa394fde64897a7b6023ad6f
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-deep_irt-ics0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 70.9, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6614663216746056, "n_train_users": 983, "n_train_chunks": 3812, "n_dev_chunks": 901, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 0.5, "flip_max_diff": 0.0, "peak_gb": 2.64}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-deep_irt-ics1/train_info.json.done b/pykt/assist2017/coldstart/pykt-deep_irt-ics1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..71070cd952e99ae42713c9a0a48efad33ae20aa0
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-deep_irt-ics1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 99.3, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6782004140309941, "n_train_users": 983, "n_train_chunks": 3759, "n_dev_chunks": 950, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 0.5, "flip_max_diff": 0.0, "peak_gb": 2.64}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-deep_irt-ics2/train_info.json.done b/pykt/assist2017/coldstart/pykt-deep_irt-ics2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..329f7854cb0fccd2420121dcf63dd5b5d06e888b
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-deep_irt-ics2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 71.4, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.675255221825015, "n_train_users": 983, "n_train_chunks": 3797, "n_dev_chunks": 935, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 0.8, "flip_max_diff": 0.0, "peak_gb": 2.64}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN128-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN128-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b977ff5e9ba10932b66dea1dd629d903f79a05c1
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN128-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 178.3, "epochs": 37, "best_epoch": 27, "dev_auc_pykt": 0.7175814492769249, "n_train_users": 102, "n_train_chunks": 526, "n_dev_chunks": 126, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 45.9, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN128-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN128-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6f209c3e0b250f59007dfbf78f99457b5123a97b
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN128-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 249.4, "epochs": 42, "best_epoch": 32, "dev_auc_pykt": 0.7053008244160841, "n_train_users": 102, "n_train_chunks": 475, "n_dev_chunks": 102, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 60.9, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN128-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN128-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..5331676ac60a3d675d0cd42eeb6520f2d1aa6391
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN128-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 152.0, "epochs": 32, "best_epoch": 22, "dev_auc_pykt": 0.6945388474800238, "n_train_users": 102, "n_train_chunks": 492, "n_dev_chunks": 139, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 49.0, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN128-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN128-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3de8c0762ab09d899406993a876a6299ab2f0985
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN128-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 232.9, "epochs": 52, "best_epoch": 42, "dev_auc_pykt": 0.710580677943879, "n_train_users": 102, "n_train_chunks": 451, "n_dev_chunks": 126, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 49.4, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN128-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN128-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..caea2b6477b6a08747d814d7f19b4abd274e5cf2
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN128-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 194.4, "epochs": 52, "best_epoch": 42, "dev_auc_pykt": 0.6795405799938021, "n_train_users": 102, "n_train_chunks": 397, "n_dev_chunks": 92, "batch_size": 64, "seq_len": 200, "n_val_users": 26, "score_s": 45.9, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN16-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN16-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7085a503ff231eb45ac8a82eb53b73b3580f0b23
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN16-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 28.5, "epochs": 43, "best_epoch": 33, "dev_auc_pykt": 0.60915046204279, "n_train_users": 13, "n_train_chunks": 56, "n_dev_chunks": 13, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 46.4, "flip_max_diff": 0.0, "peak_gb": 0.23}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN16-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN16-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6a9eb78edd568058cf07f17245da4392b8105527
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN16-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 8.3, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.6089594749388564, "n_train_users": 13, "n_train_chunks": 54, "n_dev_chunks": 13, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 48.3, "flip_max_diff": 0.0, "peak_gb": 0.22}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN16-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN16-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2c9aeddb45a32a9e072605ca50dadda919e8243c
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN16-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.5, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.6154695490469548, "n_train_users": 13, "n_train_chunks": 64, "n_dev_chunks": 14, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 48.8, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN16-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN16-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1b7d65515e951bd5ad55da74e2f471481b2ff5d4
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN16-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 17.2, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.5647230659430031, "n_train_users": 13, "n_train_chunks": 36, "n_dev_chunks": 13, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 53.0, "flip_max_diff": 0.0, "peak_gb": 0.15}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN16-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN16-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..548fe6cf657635385e22681e80c7219a50e4c515
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN16-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 11.8, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.5448906786303073, "n_train_users": 13, "n_train_chunks": 53, "n_dev_chunks": 9, "batch_size": 64, "seq_len": 200, "n_val_users": 3, "score_s": 46.9, "flip_max_diff": 0.0, "peak_gb": 0.21}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN256-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN256-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..152674e36643a012cac6286eb1ca3540c245c94d
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN256-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 268.5, "epochs": 32, "best_epoch": 22, "dev_auc_pykt": 0.729790082471228, "n_train_users": 205, "n_train_chunks": 1017, "n_dev_chunks": 216, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 45.9, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN256-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN256-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..61d7e60f85e2f0c673d0bb02ae071ce312f20ec0
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN256-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 298.3, "epochs": 31, "best_epoch": 21, "dev_auc_pykt": 0.723125103219886, "n_train_users": 205, "n_train_chunks": 954, "n_dev_chunks": 240, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 55.2, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN256-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN256-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ec09d40b62cccf148794884b68e134fed7d18e95
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN256-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 240.8, "epochs": 30, "best_epoch": 20, "dev_auc_pykt": 0.7106968447835069, "n_train_users": 205, "n_train_chunks": 955, "n_dev_chunks": 227, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 48.9, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN256-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN256-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..60a62bb32855638e4515e8f67c343f63e7de8be5
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN256-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 321.7, "epochs": 38, "best_epoch": 28, "dev_auc_pykt": 0.7239405501485944, "n_train_users": 205, "n_train_chunks": 910, "n_dev_chunks": 225, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 48.8, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN256-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN256-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..65abac389ba551312f76bd65c822439d448b75c5
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN256-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 213.8, "epochs": 28, "best_epoch": 18, "dev_auc_pykt": 0.7245086667337701, "n_train_users": 205, "n_train_chunks": 838, "n_dev_chunks": 209, "batch_size": 64, "seq_len": 200, "n_val_users": 51, "score_s": 46.9, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN32-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN32-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b62087a620c7b714f2b0f3fd5f835f181564610d
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN32-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 12.2, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.5893532713326746, "n_train_users": 26, "n_train_chunks": 103, "n_dev_chunks": 43, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 45.9, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN32-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN32-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8300f8d90bc867dd99ee5eca73d74468efad68da
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN32-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 16.8, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.596814430648092, "n_train_users": 26, "n_train_chunks": 117, "n_dev_chunks": 30, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 58.2, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN32-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN32-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..20ff2aae13e10b6c99676144ee7ce4c17d0414c6
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN32-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 19.7, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.5775792821300668, "n_train_users": 26, "n_train_chunks": 134, "n_dev_chunks": 28, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 47.8, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN32-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN32-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..d3850c35242ff17c3d8085f4acd7f6bf3e35bdd0
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN32-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 15.6, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.5929903827645943, "n_train_users": 26, "n_train_chunks": 91, "n_dev_chunks": 37, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 48.3, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN32-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN32-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a93d698758d97eb6e72fedddb473827dc6f75893
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN32-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 14.7, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.586972276865, "n_train_users": 26, "n_train_chunks": 118, "n_dev_chunks": 15, "batch_size": 64, "seq_len": 200, "n_val_users": 6, "score_s": 45.8, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN64-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN64-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..55ae2d8ff02cfa03faeb1f435fbfd04840a14920
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN64-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 136.0, "epochs": 49, "best_epoch": 39, "dev_auc_pykt": 0.6801867564108641, "n_train_users": 51, "n_train_chunks": 263, "n_dev_chunks": 71, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 46.2, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN64-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN64-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b290cc7919984417514f9b78fc17891d15b78e79
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN64-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 111.9, "epochs": 41, "best_epoch": 31, "dev_auc_pykt": 0.6504570354418733, "n_train_users": 51, "n_train_chunks": 249, "n_dev_chunks": 80, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 54.9, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN64-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN64-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e5b99fc53f4d2f30db2dbaf6783a8c83919399a6
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN64-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 93.7, "epochs": 43, "best_epoch": 33, "dev_auc_pykt": 0.6394412320965203, "n_train_users": 51, "n_train_chunks": 250, "n_dev_chunks": 54, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 52.0, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN64-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN64-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..acb30ee8697de2c251b65d1a48c5c2ef4189905f
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN64-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 27.6, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.6276771053695145, "n_train_users": 51, "n_train_chunks": 199, "n_dev_chunks": 48, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 50.3, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN64-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN64-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6d18e5c2ff0cdcfb6815ae1675a9394ed07a957d
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN64-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 135.1, "epochs": 61, "best_epoch": 51, "dev_auc_pykt": 0.6742065029582702, "n_train_users": 51, "n_train_chunks": 199, "n_dev_chunks": 57, "batch_size": 64, "seq_len": 200, "n_val_users": 13, "score_s": 47.2, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN8-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN8-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7ad1d2c53281dcc3d3bcf8c9c10f62163122e496
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN8-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 15.5, "epochs": 24, "best_epoch": 14, "dev_auc_pykt": 0.5969044815714563, "n_train_users": 6, "n_train_chunks": 28, "n_dev_chunks": 10, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 47.0, "flip_max_diff": 0.0, "peak_gb": 0.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN8-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN8-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..470b402bb9f46bf0f901ef04efafc9bd40b10243
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN8-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 42.3, "epochs": 59, "best_epoch": 49, "dev_auc_pykt": 0.5476791726791727, "n_train_users": 6, "n_train_chunks": 30, "n_dev_chunks": 6, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 48.0, "flip_max_diff": 0.0, "peak_gb": 0.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN8-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN8-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..43049e705769c252b07a2d91c6ef666d0b2c8671
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN8-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 10.5, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.5693295905722885, "n_train_users": 6, "n_train_chunks": 35, "n_dev_chunks": 7, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 53.5, "flip_max_diff": 0.0, "peak_gb": 0.14}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN8-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN8-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..336c213be00f0728d07c842b1f3f589192bc32aa
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN8-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 11.0, "epochs": 16, "best_epoch": 6, "dev_auc_pykt": 0.578029804816915, "n_train_users": 6, "n_train_chunks": 22, "n_dev_chunks": 7, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 49.6, "flip_max_diff": 0.0, "peak_gb": 0.1}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csN8-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csN8-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..296458ba8f5afcb39308c9950779fed3348e43a7
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csN8-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 14.0, "epochs": 19, "best_epoch": 9, "dev_auc_pykt": 0.5527834490268138, "n_train_users": 6, "n_train_chunks": 16, "n_dev_chunks": 7, "batch_size": 64, "seq_len": 200, "n_val_users": 2, "score_s": 50.4, "flip_max_diff": 0.0, "peak_gb": 0.1}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csNall-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csNall-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e67383e2ee71b4fb2fb203661c0f428ee10fd931
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csNall-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3639.9, "epochs": 94, "best_epoch": 84, "dev_auc_pykt": 0.7802341622130089, "n_train_users": 983, "n_train_chunks": 4659, "n_dev_chunks": 1096, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 47.8, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csNall-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csNall-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..79e431dd99360ecbddc285511348cad0fb95ac40
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csNall-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4091.8, "epochs": 100, "best_epoch": 90, "dev_auc_pykt": 0.7875081285794636, "n_train_users": 983, "n_train_chunks": 4602, "n_dev_chunks": 1153, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 52.2, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csNall-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csNall-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3fff7a1d08b0f13ffd2dc1f8d5ccb73ea7fcf88e
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csNall-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4813.5, "epochs": 115, "best_epoch": 105, "dev_auc_pykt": 0.7871799757051273, "n_train_users": 983, "n_train_chunks": 4611, "n_dev_chunks": 1144, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 51.9, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csNall-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csNall-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..4a4f07a334697f5cc2725599d23920ec3aaf111b
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csNall-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4342.0, "epochs": 100, "best_epoch": 90, "dev_auc_pykt": 0.7811910672904664, "n_train_users": 983, "n_train_chunks": 4576, "n_dev_chunks": 1179, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 56.3, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dimkt-csNall-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dimkt-csNall-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ad87c1c6850f152b7c324250ecfd554d7556deb6
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dimkt-csNall-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4353.1, "epochs": 107, "best_epoch": 97, "dev_auc_pykt": 0.7821733743944046, "n_train_users": 983, "n_train_chunks": 4613, "n_dev_chunks": 1142, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 52.6, "flip_max_diff": 0.0, "peak_gb": 0.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dkvmn-ics0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dkvmn-ics0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..73f0218a0f97ba53dfa7006a79824c3e7b583625
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dkvmn-ics0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 83.3, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6623794990886942, "n_train_users": 983, "n_train_chunks": 3812, "n_dev_chunks": 901, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 0.7, "flip_max_diff": 0.0, "peak_gb": 2.69}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dkvmn-ics1/train_info.json.done b/pykt/assist2017/coldstart/pykt-dkvmn-ics1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..fb175e657b13ab64f951a7cf39b5323d2b66b5f8
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dkvmn-ics1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 85.2, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6811154147951406, "n_train_users": 983, "n_train_chunks": 3759, "n_dev_chunks": 950, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 1.0, "flip_max_diff": 0.0, "peak_gb": 2.63}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dkvmn-ics2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dkvmn-ics2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6614375544e236d08963188387d72a3de0efeae1
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dkvmn-ics2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 78.0, "epochs": 11, "best_epoch": 1, "dev_auc_pykt": 0.6818276738406519, "n_train_users": 983, "n_train_chunks": 3797, "n_dev_chunks": 935, "batch_size": 64, "seq_len": 200, "n_val_users": 246, "score_s": 0.7, "flip_max_diff": 0.0, "peak_gb": 2.63}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN128-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN128-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1595dd79fbba82e4df2075b34259c3f5e1a10d34
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN128-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 196.3, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.7211516403359975, "n_train_users": 102, "n_train_chunks": 526, "n_dev_chunks": 126, "batch_size": 32, "seq_len": 200, "n_val_users": 26, "score_s": 61.0, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN128-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN128-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..55207e9d09609fe1b843d071b92c7a5893b83d55
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN128-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 100.9, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.6589415543480008, "n_train_users": 102, "n_train_chunks": 451, "n_dev_chunks": 126, "batch_size": 32, "seq_len": 200, "n_val_users": 26, "score_s": 38.0, "flip_max_diff": 0.0, "peak_gb": 15.18}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN128-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN128-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..6777556ad88deed497c74f8bb072ea20b98745a9
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN128-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 88.0, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.6398856215646458, "n_train_users": 102, "n_train_chunks": 397, "n_dev_chunks": 92, "batch_size": 32, "seq_len": 200, "n_val_users": 26, "score_s": 37.1, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN16-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN16-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..ec0c2d3118a3e526ab9c2e845499b730e1042da8
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN16-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 26.9, "epochs": 20, "best_epoch": 10, "dev_auc_pykt": 0.590047974036299, "n_train_users": 13, "n_train_chunks": 56, "n_dev_chunks": 13, "batch_size": 32, "seq_len": 200, "n_val_users": 3, "score_s": 58.1, "flip_max_diff": 0.0, "peak_gb": 15.15}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN16-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN16-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..9f64fb87da66d4e3b6b28cd2819e187a096e6253
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN16-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 32.3, "epochs": 24, "best_epoch": 14, "dev_auc_pykt": 0.5515651634898497, "n_train_users": 13, "n_train_chunks": 64, "n_dev_chunks": 14, "batch_size": 32, "seq_len": 200, "n_val_users": 3, "score_s": 47.5, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN16-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN16-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a70093da7f19c193bc6c458bce10af93a03f7cf4
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN16-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 14.5, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.5323639901624281, "n_train_users": 13, "n_train_chunks": 53, "n_dev_chunks": 9, "batch_size": 32, "seq_len": 200, "n_val_users": 3, "score_s": 44.8, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN256-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN256-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..d7021d196cef74d0ee5826fbdccf7c502a8e46c2
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN256-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 883.2, "epochs": 30, "best_epoch": 20, "dev_auc_pykt": 0.7185948451810855, "n_train_users": 205, "n_train_chunks": 1017, "n_dev_chunks": 216, "batch_size": 32, "seq_len": 200, "n_val_users": 51, "score_s": 58.7, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN256-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN256-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e8e6d8b06d17a20277c155f79c67034201c0c896
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN256-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 478.5, "epochs": 31, "best_epoch": 21, "dev_auc_pykt": 0.7113543071445556, "n_train_users": 205, "n_train_chunks": 910, "n_dev_chunks": 225, "batch_size": 32, "seq_len": 200, "n_val_users": 51, "score_s": 37.8, "flip_max_diff": 0.0, "peak_gb": 15.18}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN256-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN256-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8396828e3d0006bde3ada5f1ade345d3302bad99
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN256-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 695.7, "epochs": 38, "best_epoch": 28, "dev_auc_pykt": 0.72470365898419, "n_train_users": 205, "n_train_chunks": 838, "n_dev_chunks": 209, "batch_size": 32, "seq_len": 200, "n_val_users": 51, "score_s": 55.4, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN32-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN32-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..5a7a82056375ac456f3a65d7d68c0824c2a14686
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN32-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 51.9, "epochs": 20, "best_epoch": 10, "dev_auc_pykt": 0.6140479367634026, "n_train_users": 26, "n_train_chunks": 103, "n_dev_chunks": 43, "batch_size": 32, "seq_len": 200, "n_val_users": 6, "score_s": 66.7, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN32-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN32-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..acaabbe0eea04b1e5fecdc1d6ddeef887c22e121
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN32-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 57.7, "epochs": 17, "best_epoch": 7, "dev_auc_pykt": 0.6422795729624456, "n_train_users": 26, "n_train_chunks": 134, "n_dev_chunks": 28, "batch_size": 32, "seq_len": 200, "n_val_users": 6, "score_s": 61.6, "flip_max_diff": 0.0, "peak_gb": 9.35}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN32-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN32-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..454002b01ecab898ddb6d3bd94b79adc3905be8f
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN32-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 31.0, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.5577996702850362, "n_train_users": 26, "n_train_chunks": 118, "n_dev_chunks": 15, "batch_size": 32, "seq_len": 200, "n_val_users": 6, "score_s": 50.6, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN64-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN64-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..d9f22f0db8acf02ec6dfb7625743a9fc4d17481a
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN64-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 133.0, "epochs": 17, "best_epoch": 7, "dev_auc_pykt": 0.6627541266305071, "n_train_users": 51, "n_train_chunks": 263, "n_dev_chunks": 71, "batch_size": 32, "seq_len": 200, "n_val_users": 13, "score_s": 82.8, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN64-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN64-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a45946613f2a9179e678ab16d195c3668524a789
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN64-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 112.3, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.6367263132461872, "n_train_users": 51, "n_train_chunks": 250, "n_dev_chunks": 54, "batch_size": 32, "seq_len": 200, "n_val_users": 13, "score_s": 60.7, "flip_max_diff": 0.0, "peak_gb": 15.18}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN64-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN64-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b46c10c374e5c54f9a069b3deded43b734ac3a3d
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN64-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 59.1, "epochs": 17, "best_epoch": 7, "dev_auc_pykt": 0.68462956175716, "n_train_users": 51, "n_train_chunks": 199, "n_dev_chunks": 57, "batch_size": 32, "seq_len": 200, "n_val_users": 13, "score_s": 36.9, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN8-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN8-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..a5ee624b9288d5f44425717a96f0c80b7c5c74e9
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN8-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 10.6, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.5572678361178591, "n_train_users": 6, "n_train_chunks": 28, "n_dev_chunks": 10, "batch_size": 32, "seq_len": 200, "n_val_users": 2, "score_s": 57.9, "flip_max_diff": 0.0, "peak_gb": 13.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN8-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN8-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8bc86d82d5a56aac33196e792d4f30b28b81f042
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN8-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 10.9, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.5941127380320158, "n_train_users": 6, "n_train_chunks": 35, "n_dev_chunks": 7, "batch_size": 32, "seq_len": 200, "n_val_users": 2, "score_s": 49.6, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csN8-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csN8-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..08bae60ca54f54e9f92307b365e1c6a9e393ff24
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csN8-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.1, "epochs": 27, "best_epoch": 17, "dev_auc_pykt": 0.5126582278481012, "n_train_users": 6, "n_train_chunks": 16, "n_dev_chunks": 7, "batch_size": 32, "seq_len": 200, "n_val_users": 2, "score_s": 43.8, "flip_max_diff": 0.0, "peak_gb": 9.34}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csNall-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csNall-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7782a0312fa1cb44b6472ae601ef3b151202ca49
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csNall-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4204.9, "epochs": 30, "best_epoch": 20, "dev_auc_pykt": 0.7526155967862137, "n_train_users": 983, "n_train_chunks": 4659, "n_dev_chunks": 1096, "batch_size": 32, "seq_len": 200, "n_val_users": 246, "score_s": 68.0, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csNall-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csNall-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b169e9bb957327146b5ef617b9b1225dd98231fe
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csNall-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 4709.1, "epochs": 37, "best_epoch": 27, "dev_auc_pykt": 0.7654113825193052, "n_train_users": 983, "n_train_chunks": 4602, "n_dev_chunks": 1153, "batch_size": 16, "seq_len": 200, "n_val_users": 246, "score_s": 78.8, "flip_max_diff": 0.0, "peak_gb": 9.34}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-csNall-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-csNall-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..9b63584e4d0c7a209c9f224f64b281ae76f2097d
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-csNall-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3853.1, "epochs": 33, "best_epoch": 23, "dev_auc_pykt": 0.7593159568713808, "n_train_users": 983, "n_train_chunks": 4613, "n_dev_chunks": 1142, "batch_size": 16, "seq_len": 200, "n_val_users": 246, "score_s": 53.1, "flip_max_diff": 0.0, "peak_gb": 9.34}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-ics1/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-ics1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7c03623673b1f58d09a788d6ddc297fc756ae39e
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-ics1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3429.8, "epochs": 29, "best_epoch": 19, "dev_auc_pykt": 0.761635828753024, "n_train_users": 983, "n_train_chunks": 3759, "n_dev_chunks": 950, "batch_size": 32, "seq_len": 200, "n_val_users": 246, "score_s": 3.6, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-dtransformer-ics2/train_info.json.done b/pykt/assist2017/coldstart/pykt-dtransformer-ics2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..4423175a62b1977bbdcc3c2c308378668e8d083b
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-dtransformer-ics2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 3581.3, "epochs": 33, "best_epoch": 23, "dev_auc_pykt": 0.7616315932676424, "n_train_users": 983, "n_train_chunks": 3797, "n_dev_chunks": 935, "batch_size": 32, "seq_len": 200, "n_val_users": 246, "score_s": 4.1, "flip_max_diff": 0.0, "peak_gb": 15.13}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN128-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN128-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..43e7d3ee61015e336aa10283909397b22a727e21
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN128-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 16.7, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.698706594227378, "n_train_users": 102, "n_train_chunks": 526, "n_dev_chunks": 126, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 18.0, "flip_max_diff": 0.0, "peak_gb": 11.23}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN128-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN128-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..46e3e64d4c30401af8fe4960531719e103ee3c69
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN128-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 16.0, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.6990239535897004, "n_train_users": 102, "n_train_chunks": 475, "n_dev_chunks": 102, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 16.3, "flip_max_diff": 0.0, "peak_gb": 11.23}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN128-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN128-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..b8f74a3723bae2c817a33281b5cc45b5330b3458
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN128-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 19.1, "epochs": 15, "best_epoch": 5, "dev_auc_pykt": 0.635345047109753, "n_train_users": 102, "n_train_chunks": 492, "n_dev_chunks": 139, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 16.5, "flip_max_diff": 0.0, "peak_gb": 11.26}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN128-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN128-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..0ff4c30da1ce83fd7a7b3dad958191d2b34d4a39
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN128-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 9.9, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.6675286420697912, "n_train_users": 102, "n_train_chunks": 451, "n_dev_chunks": 126, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 14.5, "flip_max_diff": 0.0, "peak_gb": 11.24}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN128-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN128-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..1fb28d0885d23ec4c00ce04c1ebf61ab55dcf354
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN128-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 12.7, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.6518062750770661, "n_train_users": 102, "n_train_chunks": 397, "n_dev_chunks": 92, "batch_size": 256, "seq_len": 200, "n_val_users": 26, "score_s": 17.1, "flip_max_diff": 0.0, "peak_gb": 11.23}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN16-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN16-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e46827b4ee310bbd6c1e2bc426af396b306d7a01
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN16-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 10.0, "epochs": 20, "best_epoch": 10, "dev_auc_pykt": 0.6469905957819877, "n_train_users": 13, "n_train_chunks": 56, "n_dev_chunks": 13, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 30.2, "flip_max_diff": 0.0, "peak_gb": 2.67}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN16-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN16-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..26f97637c19be0d54482c6288a57ff49f508c30b
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN16-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.3, "epochs": 22, "best_epoch": 12, "dev_auc_pykt": 0.6530060988823875, "n_train_users": 13, "n_train_chunks": 54, "n_dev_chunks": 13, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 14.7, "flip_max_diff": 0.0, "peak_gb": 2.57}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN16-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN16-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..4409f642c61b014610fd64d72af0a101a7c67b20
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN16-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 7.2, "epochs": 32, "best_epoch": 22, "dev_auc_pykt": 0.6443224081822408, "n_train_users": 13, "n_train_chunks": 64, "n_dev_chunks": 14, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 16.4, "flip_max_diff": 0.0, "peak_gb": 3.03}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN16-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN16-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..d67d52e1136181c6515b4b1c9d8f7417e4ca80f9
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN16-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 2.9, "epochs": 14, "best_epoch": 4, "dev_auc_pykt": 0.5773761113600779, "n_train_users": 13, "n_train_chunks": 36, "n_dev_chunks": 13, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 16.2, "flip_max_diff": 0.0, "peak_gb": 1.83}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN16-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN16-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..5bc899a62df9ae731c7164c4c9cbd750dc7832bb
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN16-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 7.8, "epochs": 35, "best_epoch": 25, "dev_auc_pykt": 0.5534580146482527, "n_train_users": 13, "n_train_chunks": 53, "n_dev_chunks": 9, "batch_size": 256, "seq_len": 200, "n_val_users": 3, "score_s": 13.0, "flip_max_diff": 0.0, "peak_gb": 2.55}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN256-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN256-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..e731b11c43c4ee6c803edc7e2731a2da0f1e98db
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN256-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 28.5, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.6658307920089737, "n_train_users": 205, "n_train_chunks": 1017, "n_dev_chunks": 216, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 18.2, "flip_max_diff": 0.0, "peak_gb": 11.23}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN256-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN256-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..7a63c1d475026da14990b883a6c1b9eda327964a
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN256-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 30.4, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.6638321636276339, "n_train_users": 205, "n_train_chunks": 954, "n_dev_chunks": 240, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 19.4, "flip_max_diff": 0.0, "peak_gb": 11.24}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN256-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN256-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..78bf132e66700a85ff4c30c9cb5b5805b279e307
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN256-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 29.2, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.6338845440064769, "n_train_users": 205, "n_train_chunks": 955, "n_dev_chunks": 227, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 18.9, "flip_max_diff": 0.0, "peak_gb": 11.24}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN256-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN256-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..3f9f03ff4fe72582958731cda0e4cbb831c82ccc
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN256-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 29.6, "epochs": 13, "best_epoch": 3, "dev_auc_pykt": 0.6706740153185289, "n_train_users": 205, "n_train_chunks": 910, "n_dev_chunks": 225, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 19.0, "flip_max_diff": 0.0, "peak_gb": 11.24}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN256-s4/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN256-s4/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..22de4d76d59896b38e3b18e7cbaca5fd854259f9
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN256-s4/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 26.7, "epochs": 12, "best_epoch": 2, "dev_auc_pykt": 0.6810200450096823, "n_train_users": 205, "n_train_chunks": 838, "n_dev_chunks": 209, "batch_size": 256, "seq_len": 200, "n_val_users": 51, "score_s": 19.9, "flip_max_diff": 0.0, "peak_gb": 11.24}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN32-s0/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN32-s0/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..8f26a99d32864980e5a363119fd87e8978d3ad09
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN32-s0/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.7, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.6688044330810524, "n_train_users": 26, "n_train_chunks": 103, "n_dev_chunks": 43, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 14.8, "flip_max_diff": 0.0, "peak_gb": 4.67}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN32-s1/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN32-s1/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..2ed0e9a91b0d008809321f8d468eb56771b43f5c
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN32-s1/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 7.5, "epochs": 21, "best_epoch": 11, "dev_auc_pykt": 0.6776574803149606, "n_train_users": 26, "n_train_chunks": 117, "n_dev_chunks": 30, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 15.1, "flip_max_diff": 0.0, "peak_gb": 5.3}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN32-s2/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN32-s2/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..cb2c0d6f891e565f9f6069509bd6a5e44bc3852c
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN32-s2/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 10.5, "epochs": 25, "best_epoch": 15, "dev_auc_pykt": 0.6295053810221393, "n_train_users": 26, "n_train_chunks": 134, "n_dev_chunks": 28, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 14.8, "flip_max_diff": 0.0, "peak_gb": 6.01}
\ No newline at end of file
diff --git a/pykt/assist2017/coldstart/pykt-extrakt-csN32-s3/train_info.json.done b/pykt/assist2017/coldstart/pykt-extrakt-csN32-s3/train_info.json.done
new file mode 100644
index 0000000000000000000000000000000000000000..db49bf24d60850a595f0d5bf7aa5842f4e8e6318
--- /dev/null
+++ b/pykt/assist2017/coldstart/pykt-extrakt-csN32-s3/train_info.json.done
@@ -0,0 +1 @@
+{"train_s": 6.0, "epochs": 18, "best_epoch": 8, "dev_auc_pykt": 0.6604519550691561, "n_train_users": 26, "n_train_chunks": 91, "n_dev_chunks": 37, "batch_size": 256, "seq_len": 200, "n_val_users": 6, "score_s": 16.9, "flip_max_diff": 0.0, "peak_gb": 4.16}
\ No newline at end of file