View training loss curve.

examples/Model definition

@@ -235,7 +235,7 @@
       "id": "Input: tensor 3",
       "position": {
         "x": 485.8840220312055,
-        "y": -149.86223034126274
+        "y": -268.0485936515193
       },
       "type": "basic",
       "width": 200.0
@@ -285,8 +285,8 @@
       "height": 200.0,
       "id": "MSE loss 2",
       "position": {
-        "x": 362.77557479979805,
-        "y": -1287.1056805642488
+        "x": 384.54674698852955,
+        "y": -1184.4701545316577
       },
       "type": "basic",
       "width": 200.0
@@ -513,8 +513,8 @@
       "height": 258.0,
       "id": "Constant vector 1",
       "position": {
-        "x": 983.1241140187901,
-        "y": -562.803650462906
+        "x": 886.708922897265,
+        "y": -298.4394167425953
       },
       "type": "basic",
       "width": 238.0
@@ -568,8 +568,8 @@
       "height": 200.0,
       "id": "Add 1",
       "position": {
-        "x": 818.5444381090571,
-        "y": -955.5157374399466
+        "x": 722.1292469875319,
+        "y": -762.6853551968964
       },
       "type": "basic",
       "width": 200.0
@@ -621,6 +621,59 @@
       },
       "type": "basic",
       "width": 200.0
+    },
+    {
+      "data": {
+        "display": null,
+        "error": null,
+        "input_metadata": null,
+        "meta": {
+          "inputs": {},
+          "name": "Constant vector",
+          "outputs": {
+            "output": {
+              "name": "output",
+              "position": "top",
+              "type": {
+                "type": "None"
+              }
+            }
+          },
+          "params": {
+            "size": {
+              "default": 1.0,
+              "name": "size",
+              "type": {
+                "type": "<class 'int'>"
+              }
+            },
+            "value": {
+              "default": 0.0,
+              "name": "value",
+              "type": {
+                "type": "<class 'int'>"
+              }
+            }
+          },
+          "position": {
+            "x": 318.0,
+            "y": 215.0
+          },
+          "type": "basic"
+        },
+        "params": {
+          "size": 1.0,
+          "value": 0.0
+        },
+        "status": "done",
+        "title": "Constant vector"
+      },
+      "id": "Constant vector 2",
+      "position": {
+        "x": -1125.0,
+        "y": -780.0
+      },
+      "type": "basic"
     }
   ]
 }

examples/Model use

@@ -34,6 +34,13 @@
       "sourceHandle": "output",
       "target": "View tables 1",
       "targetHandle": "bundle"
+    },
+    {
+      "id": "Train model 2 View loss 1",
+      "source": "Train model 2",
+      "sourceHandle": "output",
+      "target": "View loss 1",
+      "targetHandle": "bundle"
     }
   ],
   "env": "LynxKite Graph Analytics",
@@ -579,54 +586,54 @@
               ],
               "data": [
                 [
-                  "[0.56922203 0.98222166 0.76851749 0.28615737]",
-                  "[1.56922197 1.9822216  1.76851749 1.28615737]",
-                  "[2.5075035095214844, 3.0133981704711914, 2.698194980621338, 2.3600802421569824]"
+                  "[4.27091718e-01 4.89909172e-01 6.92297399e-01 2.57611275e-04]",
+                  "[1.42709172 1.48990917 1.69229746 1.00025761]",
+                  "[2.3393754959106445, 2.3441734313964844, 2.4766101837158203, 2.484668731689453]"
                 ],
                 [
-                  "[0.98033333 0.97656083 0.38939917 0.81491041]",
-                  "[1.98033333 1.97656083 1.38939917 1.81491041]",
-                  "[3.0080792903900146, 2.9657773971557617, 2.4899187088012695, 2.7300877571105957]"
+                  "[0.75292218 0.81470108 0.49657214 0.56217098]",
+                  "[1.75292218 1.81470108 1.49657214 1.56217098]",
+                  "[2.7774970531463623, 2.769310712814331, 2.456480026245117, 2.6388649940490723]"
                 ],
                 [
-                  "[0.74064726 0.4155122  0.09800029 0.49930882]",
-                  "[1.74064732 1.4155122  1.09800029 1.49930882]",
-                  "[2.7571821212768555, 2.4066381454467773, 2.099902629852295, 2.491953134536743]"
+                  "[0.72290605 0.96945059 0.68354797 0.15270454]",
+                  "[1.72290611 1.96945059 1.68354797 1.15270448]",
+                  "[2.751509189605713, 2.74546480178833, 2.446249485015869, 2.624647855758667]"
                 ],
                 [
-                  "[0.68891573 0.25576538 0.96339929 0.503833  ]",
-                  "[1.68891573 1.25576544 1.96339929 1.50383306]",
-                  "[2.4299726486206055, 2.4633498191833496, 2.669276237487793, 2.753784656524658]"
+                  "[0.9630242  0.76359051 0.24853623 0.76881069]",
+                  "[1.96302414 1.76359057 1.24853623 1.76881075]",
+                  "[2.888108253479004, 2.9004437923431396, 2.349879026412964, 2.634594202041626]"
                 ],
                 [
-                  "[0.87285906 0.48354989 0.39394957 0.59456545]",
-                  "[1.872859   1.48354983 1.39394951 1.59456539]",
-                  "[2.7922420501708984, 2.555819034576416, 2.3168320655822754, 2.6561851501464844]"
+                  "[0.67418337 0.79634351 0.23229051 0.71345252]",
+                  "[1.67418337 1.79634356 1.23229051 1.71345258]",
+                  "[2.7448184490203857, 2.7737462520599365, 2.300586223602295, 2.5605084896087646]"
                 ],
                 [
-                  "[0.32565445 0.90939188 0.07488042 0.13730896]",
-                  "[1.32565451 1.90939188 1.07488036 1.13730896]",
-                  "[2.5029656887054443, 2.744194984436035, 2.2190115451812744, 2.024731159210205]"
+                  "[0.98033333 0.97656083 0.38939917 0.81491041]",
+                  "[1.98033333 1.97656083 1.38939917 1.81491041]",
+                  "[3.020817995071411, 3.0079593658447266, 2.4341537952423096, 2.7197117805480957]"
                 ],
                 [
-                  "[0.49607176 0.1922397  0.46640229 0.78321403]",
-                  "[1.49607182 1.19223976 1.46640229 1.78321409]",
-                  "[2.4844861030578613, 2.2119431495666504, 2.451040267944336, 2.7876930236816406]"
+                  "[0.49584109 0.80599248 0.07096875 0.75872749]",
+                  "[1.49584103 1.80599248 1.07096875 1.75872755]",
+                  "[2.6583402156829834, 2.7179207801818848, 2.1810216903686523, 2.477621078491211]"
                 ],
                 [
-                  "[0.85566247 0.83362883 0.48424995 0.25265992]",
-                  "[1.85566247 1.83362889 1.48424995 1.25265992]",
-                  "[2.757845401763916, 2.9018521308898926, 2.371169328689575, 2.356513500213623]"
+                  "[0.30631393 0.48311198 0.87847513 0.67559886]",
+                  "[1.30631399 1.48311198 1.87847519 1.67559886]",
+                  "[2.455927848815918, 2.3795063495635986, 2.8169684410095215, 2.673161506652832]"
                 ],
                 [
-                  "[0.48959708 0.48549271 0.32688856 0.356677  ]",
-                  "[1.48959708 1.48549271 1.32688856 1.35667706]",
-                  "[2.4994757175445557, 2.469498634338379, 2.3010873794555664, 2.3796985149383545]"
+                  "[0.04508126 0.76880038 0.80721325 0.62542385]",
+                  "[1.04508126 1.76880038 1.80721331 1.62542391]",
+                  "[2.443504810333252, 2.396592855453491, 2.6842336654663086, 2.612111806869507]"
                 ],
                 [
-                  "[0.80654246 0.08253473 0.74478531 0.71257162]",
-                  "[1.8065424  1.08253479 1.74478531 1.71257162]",
-                  "[2.5831751823425293, 2.2751128673553467, 2.5047459602355957, 2.9079394340515137]"
+                  "[0.77427191 0.21829212 0.12769502 0.74303615]",
+                  "[1.77427197 1.21829212 1.12769508 1.74303615]",
+                  "[2.5184009075164795, 2.5502569675445557, 2.3369357585906982, 2.4919686317443848]"
                 ]
               ]
             },
@@ -668,10 +675,6 @@
                   "[0.18149549 0.30520517 0.30946714 0.16786289]",
                   "[1.18149543 1.30520511 1.30946708 1.16786289]"
                 ],
-                [
-                  "[4.27091718e-01 4.89909172e-01 6.92297399e-01 2.57611275e-04]",
-                  "[1.42709172 1.48990917 1.69229746 1.00025761]"
-                ],
                 [
                   "[0.32225502 0.16999388 0.05823922 0.9628762 ]",
                   "[1.32225502 1.16999388 1.05823922 1.9628762 ]"
@@ -689,12 +692,12 @@
                   "[1.11693287 1.49860179 1.55020833 1.88832855]"
                 ],
                 [
-                  "[0.50272274 0.54912758 0.17663097 0.79070699]",
-                  "[1.50272274 1.54912758 1.17663097 1.79070699]"
+                  "[0.48959708 0.48549271 0.32688856 0.356677  ]",
+                  "[1.48959708 1.48549271 1.32688856 1.35667706]"
                 ],
                 [
-                  "[0.04508126 0.76880038 0.80721325 0.62542385]",
-                  "[1.04508126 1.76880038 1.80721331 1.62542391]"
+                  "[0.50272274 0.54912758 0.17663097 0.79070699]",
+                  "[1.50272274 1.54912758 1.17663097 1.79070699]"
                 ],
                 [
                   "[0.19908059 0.17570406 0.51475513 0.1893943 ]",
@@ -716,6 +719,10 @@
                   "[0.62569475 0.9881897  0.83639616 0.9828859 ]",
                   "[1.62569475 1.9881897  1.83639622 1.98288584]"
                 ],
+                [
+                  "[0.56922203 0.98222166 0.76851749 0.28615737]",
+                  "[1.56922197 1.9822216  1.76851749 1.28615737]"
+                ],
                 [
                   "[0.88776821 0.51636773 0.30333066 0.32230979]",
                   "[1.88776827 1.51636767 1.30333066 1.32230973]"
@@ -768,6 +775,10 @@
                   "[0.4827103  0.10563457 0.98858833 0.82286644]",
                   "[1.48271036 1.10563457 1.98858833 1.82286644]"
                 ],
+                [
+                  "[0.74064726 0.4155122  0.09800029 0.49930882]",
+                  "[1.74064732 1.4155122  1.09800029 1.49930882]"
+                ],
                 [
                   "[0.78956431 0.87284744 0.06880784 0.03455889]",
                   "[1.78956437 1.87284744 1.06880784 1.03455889]"
@@ -784,14 +795,6 @@
                   "[0.44330525 0.09997386 0.89025736 0.90507984]",
                   "[1.44330525 1.09997392 1.89025736 1.90507984]"
                 ],
-                [
-                  "[0.72290605 0.96945059 0.68354797 0.15270454]",
-                  "[1.72290611 1.96945059 1.68354797 1.15270448]"
-                ],
-                [
-                  "[0.75292218 0.81470108 0.49657214 0.56217098]",
-                  "[1.75292218 1.81470108 1.49657214 1.56217098]"
-                ],
                 [
                   "[0.33480108 0.59181517 0.76198453 0.98062384]",
                   "[1.33480108 1.59181523 1.76198459 1.98062384]"
@@ -812,10 +815,6 @@
                   "[0.60110539 0.3618983  0.32342511 0.98672163]",
                   "[1.60110545 1.3618983  1.32342505 1.98672163]"
                 ],
-                [
-                  "[0.77427191 0.21829212 0.12769502 0.74303615]",
-                  "[1.77427197 1.21829212 1.12769508 1.74303615]"
-                ],
                 [
                   "[0.08107251 0.2602725  0.18861133 0.44833237]",
                   "[1.08107257 1.2602725  1.18861127 1.44833231]"
@@ -828,18 +827,22 @@
                   "[0.93488538 0.73882395 0.37345302 0.0274905 ]",
                   "[1.93488538 1.73882389 1.37345302 1.0274905 ]"
                 ],
-                [
-                  "[0.30631393 0.48311198 0.87847513 0.67559886]",
-                  "[1.30631399 1.48311198 1.87847519 1.67559886]"
-                ],
                 [
                   "[0.18720162 0.74115586 0.98626411 0.30355608]",
                   "[1.18720162 1.74115586 1.98626411 1.30355608]"
                 ],
+                [
+                  "[0.85566247 0.83362883 0.48424995 0.25265992]",
+                  "[1.85566247 1.83362889 1.48424995 1.25265992]"
+                ],
                 [
                   "[0.95928186 0.84273899 0.71514636 0.38619852]",
                   "[1.95928192 1.84273899 1.7151463  1.38619852]"
                 ],
+                [
+                  "[0.32565445 0.90939188 0.07488042 0.13730896]",
+                  "[1.32565451 1.90939188 1.07488036 1.13730896]"
+                ],
                 [
                   "[0.9829582  0.59269661 0.40120947 0.95487177]",
                   "[1.9829582  1.59269667 1.40120947 1.95487177]"
@@ -853,8 +856,8 @@
                   "[1.54914117 1.03810108 1.87531948 1.73044229]"
                 ],
                 [
-                  "[0.67418337 0.79634351 0.23229051 0.71345252]",
-                  "[1.67418337 1.79634356 1.23229051 1.71345258]"
+                  "[0.87285906 0.48354989 0.39394957 0.59456545]",
+                  "[1.872859   1.48354983 1.39394951 1.59456539]"
                 ],
                 [
                   "[0.81788456 0.58174163 0.29376316 0.7971254 ]",
@@ -892,14 +895,14 @@
                   "[0.47963417 0.81818312 0.48720706 0.49339259]",
                   "[1.47963417 1.81818318 1.48720706 1.49339259]"
                 ],
-                [
-                  "[0.9630242  0.76359051 0.24853623 0.76881069]",
-                  "[1.96302414 1.76359057 1.24853623 1.76881075]"
-                ],
                 [
                   "[0.60609657 0.96257663 0.19292736 0.95702219]",
                   "[1.60609651 1.96257663 1.19292736 1.95702219]"
                 ],
+                [
+                  "[0.80654246 0.08253473 0.74478531 0.71257162]",
+                  "[1.8065424  1.08253479 1.74478531 1.71257162]"
+                ],
                 [
                   "[0.70167565 0.26930219 0.5660674  0.61194974]",
                   "[1.70167565 1.26930213 1.56606746 1.61194968]"
@@ -940,6 +943,10 @@
                   "[0.37959969 0.42820001 0.10690689 0.96353984]",
                   "[1.37959969 1.42820001 1.10690689 1.96353984]"
                 ],
+                [
+                  "[0.49607176 0.1922397  0.46640229 0.78321403]",
+                  "[1.49607182 1.19223976 1.46640229 1.78321409]"
+                ],
                 [
                   "[0.40234613 0.54987347 0.49542785 0.54153186]",
                   "[1.40234613 1.54987347 1.49542785 1.5415318 ]"
@@ -960,10 +967,6 @@
                   "[0.47856545 0.46267092 0.6376707  0.84747767]",
                   "[1.47856545 1.46267092 1.63767076 1.84747767]"
                 ],
-                [
-                  "[0.49584109 0.80599248 0.07096875 0.75872749]",
-                  "[1.49584103 1.80599248 1.07096875 1.75872755]"
-                ],
                 [
                   "[0.43500566 0.66041756 0.80293626 0.96224713]",
                   "[1.43500566 1.66041756 1.80293632 1.96224713]"
@@ -995,12 +998,323 @@
                 [
                   "[0.87608397 0.93200487 0.80169648 0.37758952]",
                   "[1.87608397 1.93200493 1.80169654 1.37758946]"
+                ],
+                [
+                  "[0.68891573 0.25576538 0.96339929 0.503833  ]",
+                  "[1.68891573 1.25576544 1.96339929 1.50383306]"
+                ]
+              ]
+            },
+            "training": {
+              "columns": [
+                "training_loss"
+              ],
+              "data": [
+                [
+                  5.919814109802246
+                ],
+                [
+                  5.590979099273682
+                ],
+                [
+                  5.038528919219971
+                ],
+                [
+                  4.499487400054932
+                ],
+                [
+                  4.035862445831299
+                ],
+                [
+                  3.6894421577453613
+                ],
+                [
+                  3.472367763519287
+                ],
+                [
+                  3.3581743240356445
+                ],
+                [
+                  3.296724557876587
+                ],
+                [
+                  3.124946117401123
+                ],
+                [
+                  2.452446460723877
+                ],
+                [
+                  2.0572497844696045
+                ],
+                [
+                  1.8718140125274658
+                ],
+                [
+                  1.7930662631988525
+                ],
+                [
+                  1.7607650756835938
+                ],
+                [
+                  1.7473256587982178
+                ],
+                [
+                  1.7414778470993042
+                ],
+                [
+                  1.738724946975708
+                ],
+                [
+                  1.737252116203308
+                ],
+                [
+                  1.7363152503967285
+                ],
+                [
+                  1.7356064319610596
+                ],
+                [
+                  1.7349969148635864
+                ],
+                [
+                  1.7344319820404053
+                ],
+                [
+                  1.7338895797729492
+                ],
+                [
+                  1.7333598136901855
+                ],
+                [
+                  1.7328382730484009
+                ],
+                [
+                  1.7323238849639893
+                ],
+                [
+                  1.731815218925476
+                ],
+                [
+                  1.7313120365142822
+                ],
+                [
+                  1.7308142185211182
+                ],
+                [
+                  1.7303216457366943
+                ],
+                [
+                  1.7298336029052734
+                ],
+                [
+                  1.7293508052825928
+                ],
+                [
+                  1.7288726568222046
+                ],
+                [
+                  1.7283992767333984
+                ],
+                [
+                  1.7279306650161743
+                ],
+                [
+                  1.727466344833374
+                ],
+                [
+                  1.7270063161849976
+                ],
+                [
+                  1.7265511751174927
+                ],
+                [
+                  1.7260998487472534
+                ],
+                [
+                  1.7256529331207275
+                ],
+                [
+                  1.7252100706100464
+                ],
+                [
+                  1.7247711420059204
+                ],
+                [
+                  1.7243363857269287
+                ],
+                [
+                  1.7239054441452026
+                ],
+                [
+                  1.7234781980514526
+                ],
+                [
+                  1.7230546474456787
+                ],
+                [
+                  1.722635269165039
+                ],
+                [
+                  1.7222192287445068
+                ],
+                [
+                  1.7218068838119507
+                ],
+                [
+                  1.7213975191116333
+                ],
+                [
+                  1.7209922075271606
+                ],
+                [
+                  1.7205902338027954
+                ],
+                [
+                  1.7201911211013794
+                ],
+                [
+                  1.7197957038879395
+                ],
+                [
+                  1.7194033861160278
+                ],
+                [
+                  1.7190141677856445
+                ],
+                [
+                  1.7186282873153687
+                ],
+                [
+                  1.7182451486587524
+                ],
+                [
+                  1.7178229093551636
+                ],
+                [
+                  1.7170121669769287
+                ],
+                [
+                  1.712469458580017
+                ],
+                [
+                  1.560640573501587
+                ],
+                [
+                  0.777249813079834
+                ],
+                [
+                  0.3986421227455139
+                ],
+                [
+                  0.21910731494426727
+                ],
+                [
+                  0.13873876631259918
+                ],
+                [
+                  0.10362569987773895
+                ],
+                [
+                  0.08824621140956879
+                ],
+                [
+                  0.0813695564866066
+                ],
+                [
+                  0.07816052436828613
+                ],
+                [
+                  0.07654092460870743
+                ],
+                [
+                  0.07561471313238144
+                ],
+                [
+                  0.0749940350651741
+                ],
+                [
+                  0.07451044768095016
+                ],
+                [
+                  0.0740906149148941
+                ],
+                [
+                  0.0737023800611496
+                ],
+                [
+                  0.07333160191774368
+                ],
+                [
+                  0.07297196984291077
+                ],
+                [
+                  0.07262063771486282
+                ],
+                [
+                  0.07227623462677002
+                ],
+                [
+                  0.07193806767463684
+                ],
+                [
+                  0.07160572707653046
+                ],
+                [
+                  0.07127895951271057
+                ],
+                [
+                  0.07095754891633987
+                ],
+                [
+                  0.07064132392406464
+                ],
+                [
+                  0.07033013552427292
+                ],
+                [
+                  0.07002381980419159
+                ],
+                [
+                  0.0697222352027893
+                ],
+                [
+                  0.06942524015903473
+                ],
+                [
+                  0.0691327303647995
+                ],
+                [
+                  0.06884456425905228
+                ],
+                [
+                  0.06856060773134232
+                ],
+                [
+                  0.06828076392412186
+                ],
+                [
+                  0.06800491362810135
+                ],
+                [
+                  0.06773293763399124
+                ],
+                [
+                  0.06746473908424377
+                ],
+                [
+                  0.06720020622015
+                ],
+                [
+                  0.06693925708532333
+                ],
+                [
+                  0.06668176501989365
                 ]
               ]
             }
           },
           "other": {
-            "model": "ModelConfig(model=Sequential(\n  (0) - Identity(): Input__tensor_1_output -> START_Repeat_1_output\n  (1) - Linear(4, 4, bias=True): START_Repeat_1_output -> Linear_1_output\n  (2) - <function leaky_relu at 0x7b4a644509a0>: Linear_1_output -> Activation_1_output\n  (3) - Identity(): Activation_1_output -> START_Repeat_1_output\n  (4) - Linear(4, 4, bias=True): START_Repeat_1_output -> Linear_1_output\n  (5) - <function leaky_relu at 0x7b4a644509a0>: Linear_1_output -> Activation_1_output\n  (6) - Identity(): Activation_1_output -> END_Repeat_1_output\n  (7) - Identity(): END_Repeat_1_output -> Output_1_x\n  (8) - Identity(): Output_1_x -> Output_1_x\n), model_inputs=['Input__tensor_1_output'], model_outputs=['Output_1_x'], loss_inputs=['Input__tensor_3_output', 'Output_1_x'], loss=Sequential(\n  (0) - <function constant_vector.<locals>.<lambda> at 0x7b4975350e00>: nothing -> Constant_vector_1_output\n  (1) - <built-in method add of type object at 0x7b4a5859ef00>: Input__tensor_3_output, Constant_vector_1_output -> Add_1_output\n  (2) - <function mse_loss at 0x7b4a64452480>: Output_1_x, Add_1_output -> MSE_loss_2_output\n  (3) - Identity(): MSE_loss_2_output -> loss\n), optimizer_parameters={'lr': 0.1, 'type': <OptionsFor_type.SGD: 4>}, optimizer=SGD (\nParameter Group 0\n    dampening: 0\n    differentiable: False\n    foreach: None\n    fused: None\n    lr: 0.1\n    maximize: False\n    momentum: 0\n    nesterov: False\n    weight_decay: 0\n), source_workspace='Model definition', trained=True)"
+            "model": "ModelConfig(model=Sequential(\n  (0) - Identity(): Input__tensor_1_output -> START_Repeat_1_output\n  (1) - Linear(4, 4, bias=True): START_Repeat_1_output -> Linear_1_output\n  (2) - <function leaky_relu at 0x7e6bbbf30a40>: Linear_1_output -> Activation_1_output\n  (3) - Identity(): Activation_1_output -> START_Repeat_1_output\n  (4) - Linear(4, 4, bias=True): START_Repeat_1_output -> Linear_1_output\n  (5) - <function leaky_relu at 0x7e6bbbf30a40>: Linear_1_output -> Activation_1_output\n  (6) - Identity(): Activation_1_output -> END_Repeat_1_output\n  (7) - Identity(): END_Repeat_1_output -> Output_1_x\n  (8) - Identity(): Output_1_x -> Output_1_x\n), model_inputs=['Input__tensor_1_output'], model_outputs=['Output_1_x'], loss_inputs=['Input__tensor_3_output', 'Output_1_x'], loss=Sequential(\n  (0) - <function constant_vector.<locals>.<lambda> at 0x7e6adc9280e0>: nothing -> Constant_vector_1_output\n  (1) - <built-in method add of type object at 0x7e6bb459ef00>: Input__tensor_3_output, Constant_vector_1_output -> Add_1_output\n  (2) - <function mse_loss at 0x7e6bbbf32520>: Output_1_x, Add_1_output -> MSE_loss_2_output\n  (3) - Identity(): MSE_loss_2_output -> loss\n), optimizer_parameters={'lr': 0.1, 'type': <OptionsFor_type.SGD: 4>}, optimizer=SGD (\nParameter Group 0\n    dampening: 0\n    differentiable: False\n    foreach: None\n    fused: None\n    lr: 0.1\n    maximize: False\n    momentum: 0\n    nesterov: False\n    weight_decay: 0\n), source_workspace='Model definition', trained=True)"
           },
           "relations": []
         },
@@ -1026,6 +1340,11 @@
                   "x",
                   "y"
                 ]
+              },
+              "training": {
+                "columns": [
+                  "training_loss"
+                ]
               }
             },
             "other": {
@@ -1270,7 +1589,7 @@
           "type": "basic"
         },
         "params": {
-          "epochs": "1001",
+          "epochs": "500",
           "input_mapping": "{\"map\":{\"Input__tensor_1_output\":{\"df\":\"df_train\",\"column\":\"x\"},\"Input__tensor_3_output\":{\"df\":\"df_train\",\"column\":\"y\"}}}",
           "model_name": "model"
         },
@@ -1313,6 +1632,11 @@
                   "x",
                   "y"
                 ]
+              },
+              "training": {
+                "columns": [
+                  "training_loss"
+                ]
               }
             },
             "other": {
@@ -1398,6 +1722,610 @@
       },
       "type": "basic",
       "width": 529.0
+    },
+    {
+      "data": {
+        "__execution_delay": null,
+        "collapsed": false,
+        "display": {
+          "series": [
+            {
+              "data": [
+                5.919814109802246,
+                5.590979099273682,
+                5.038528919219971,
+                4.499487400054932,
+                4.035862445831299,
+                3.6894421577453613,
+                3.472367763519287,
+                3.3581743240356445,
+                3.296724557876587,
+                3.124946117401123,
+                2.452446460723877,
+                2.0572497844696045,
+                1.8718140125274658,
+                1.7930662631988525,
+                1.7607650756835938,
+                1.7473256587982178,
+                1.7414778470993042,
+                1.738724946975708,
+                1.737252116203308,
+                1.7363152503967285,
+                1.7356064319610596,
+                1.7349969148635864,
+                1.7344319820404053,
+                1.7338895797729492,
+                1.7333598136901855,
+                1.7328382730484009,
+                1.7323238849639893,
+                1.731815218925476,
+                1.7313120365142822,
+                1.7308142185211182,
+                1.7303216457366943,
+                1.7298336029052734,
+                1.7293508052825928,
+                1.7288726568222046,
+                1.7283992767333984,
+                1.7279306650161743,
+                1.727466344833374,
+                1.7270063161849976,
+                1.7265511751174927,
+                1.7260998487472534,
+                1.7256529331207275,
+                1.7252100706100464,
+                1.7247711420059204,
+                1.7243363857269287,
+                1.7239054441452026,
+                1.7234781980514526,
+                1.7230546474456787,
+                1.722635269165039,
+                1.7222192287445068,
+                1.7218068838119507,
+                1.7213975191116333,
+                1.7209922075271606,
+                1.7205902338027954,
+                1.7201911211013794,
+                1.7197957038879395,
+                1.7194033861160278,
+                1.7190141677856445,
+                1.7186282873153687,
+                1.7182451486587524,
+                1.7178229093551636,
+                1.7170121669769287,
+                1.712469458580017,
+                1.560640573501587,
+                0.777249813079834,
+                0.3986421227455139,
+                0.21910731494426727,
+                0.13873876631259918,
+                0.10362569987773895,
+                0.08824621140956879,
+                0.0813695564866066,
+                0.07816052436828613,
+                0.07654092460870743,
+                0.07561471313238144,
+                0.0749940350651741,
+                0.07451044768095016,
+                0.0740906149148941,
+                0.0737023800611496,
+                0.07333160191774368,
+                0.07297196984291077,
+                0.07262063771486282,
+                0.07227623462677002,
+                0.07193806767463684,
+                0.07160572707653046,
+                0.07127895951271057,
+                0.07095754891633987,
+                0.07064132392406464,
+                0.07033013552427292,
+                0.07002381980419159,
+                0.0697222352027893,
+                0.06942524015903473,
+                0.0691327303647995,
+                0.06884456425905228,
+                0.06856060773134232,
+                0.06828076392412186,
+                0.06800491362810135,
+                0.06773293763399124,
+                0.06746473908424377,
+                0.06720020622015,
+                0.06693925708532333,
+                0.06668176501989365,
+                0.06642767041921616,
+                0.0661768838763237,
+                0.06592927873134613,
+                0.06568481773138046,
+                0.06544338166713715,
+                0.0652049258351326,
+                0.06496933847665787,
+                0.06473657488822937,
+                0.06450657546520233,
+                0.0642792209982872,
+                0.06405448168516159,
+                0.06383229047060013,
+                0.06361258029937744,
+                0.06339527666568756,
+                0.0631803572177887,
+                0.06296773254871368,
+                0.06275734305381775,
+                0.0625491589307785,
+                0.06234312057495117,
+                0.06213918328285217,
+                0.06193728372454643,
+                0.06173737347126007,
+                0.061539426445961,
+                0.06134339049458504,
+                0.06114921718835831,
+                0.06095685437321663,
+                0.06076629459857941,
+                0.06057746708393097,
+                0.06039035692811012,
+                0.06020490825176239,
+                0.06002110242843628,
+                0.05983888357877731,
+                0.059658247977495193,
+                0.05947912484407425,
+                0.05930151790380478,
+                0.0591253824532032,
+                0.05895068868994713,
+                0.05877740681171417,
+                0.058605507016181946,
+                0.058434974402189255,
+                0.05826576426625252,
+                0.05809786170721054,
+                0.05793124437332153,
+                0.05776588246226311,
+                0.05760175362229347,
+                0.057438839226961136,
+                0.057277124375104904,
+                0.0571165569126606,
+                0.056957144290208817,
+                0.05679886043071747,
+                0.05664169043302536,
+                0.056485600769519806,
+                0.056330591440200806,
+                0.05617662891745567,
+                0.05602370947599411,
+                0.05587179958820343,
+                0.055720895528793335,
+                0.05557098239660263,
+                0.05542204529047012,
+                0.05527406185865402,
+                0.05512702092528343,
+                0.05498090386390686,
+                0.05483569949865341,
+                0.054691411554813385,
+                0.0545479990541935,
+                0.05440546199679375,
+                0.054263804107904434,
+                0.054122988134622574,
+                0.053983014076948166,
+                0.05384387448430061,
+                0.05370555818080902,
+                0.053568046540021896,
+                0.05343133956193924,
+                0.05329542234539986,
+                0.05316028743982315,
+                0.05302591249346733,
+                0.05289231985807419,
+                0.05275947228074074,
+                0.05262737348675728,
+                0.05249601602554321,
+                0.05236538127064705,
+                0.052235476672649384,
+                0.052106283605098724,
+                0.05197780951857567,
+                0.051850032061338425,
+                0.051722943782806396,
+                0.051596563309431076,
+                0.051470860838890076,
+                0.0513458251953125,
+                0.05122147127985954,
+                0.051097773015499115,
+                0.05097474530339241,
+                0.05085237696766853,
+                0.05073065683245659,
+                0.050609588623046875,
+                0.0504891499876976,
+                0.05036935955286026,
+                0.05025019496679306,
+                0.0501316636800766,
+                0.05001375824213028,
+                0.0498964786529541,
+                0.049779802560806274,
+                0.04966374486684799,
+                0.04954829812049866,
+                0.04943346977233887,
+                0.04931923374533653,
+                0.04920560494065285,
+                0.04909256845712662,
+                0.04898013174533844,
+                0.048868291079998016,
+                0.048757024109363556,
+                0.048646364361047745,
+                0.0485362783074379,
+                0.04842677712440491,
+                0.048317860811948776,
+                0.04820951819419861,
+                0.048101749271154404,
+                0.04799456149339676,
+                0.04788793995976448,
+                0.04778188467025757,
+                0.04767640307545662,
+                0.04757149517536163,
+                0.04746714234352112,
+                0.047363366931676865,
+                0.04726014286279678,
+                0.04715748876333237,
+                0.04705539345741272,
+                0.046953849494457245,
+                0.04685286059975624,
+                0.0467524379491806,
+                0.04665256664156914,
+                0.04655325040221214,
+                0.04645449295639992,
+                0.04635626822710037,
+                0.04625860974192619,
+                0.046161502599716187,
+                0.04606493189930916,
+                0.0459689237177372,
+                0.045873455703258514,
+                0.0457785464823246,
+                0.04568416252732277,
+                0.04559033736586571,
+                0.04549706354737282,
+                0.04540431872010231,
+                0.04531212896108627,
+                0.045220475643873215,
+                0.04512936249375343,
+                0.045038796961307526,
+                0.0449487678706646,
+                0.044859275221824646,
+                0.04477033019065857,
+                0.04468192160129547,
+                0.04459404945373535,
+                0.04450671374797821,
+                0.04441992565989494,
+                0.044333670288324356,
+                0.04424794763326645,
+                0.04416275396943092,
+                0.04407810792326927,
+                0.04399399086833,
+                0.04391040652990341,
+                0.0438273623585701,
+                0.04374483972787857,
+                0.04366285726428032,
+                0.043581414967775345,
+                0.04350048676133156,
+                0.04342009127140045,
+                0.043340232223272324,
+                0.043260905891656876,
+                0.04318211227655411,
+                0.04310382902622223,
+                0.04302608594298363,
+                0.04294886812567711,
+                0.042872171849012375,
+                0.042796000838279724,
+                0.04272035136818886,
+                0.042645230889320374,
+                0.042570631951093674,
+                0.04249655827879906,
+                0.04242299869656563,
+                0.042349960654973984,
+                0.04227744787931442,
+                0.04220544546842575,
+                0.042133964598178864,
+                0.04206300154328346,
+                0.04199255257844925,
+                0.04192262142896652,
+                0.04185319319367409,
+                0.04178428277373314,
+                0.04171588271856308,
+                0.04164800047874451,
+                0.04158061742782593,
+                0.041513748466968536,
+                0.041447386145591736,
+                0.04138151928782463,
+                0.04131617024540901,
+                0.04125131666660309,
+                0.04118696227669716,
+                0.04112311080098152,
+                0.04105975478887558,
+                0.04099690169095993,
+                0.04093454033136368,
+                0.04087267071008682,
+                0.04081130400300026,
+                0.040750421583652496,
+                0.04069003090262413,
+                0.040630124509334564,
+                0.0405707024037838,
+                0.040511779487133026,
+                0.04045332968235016,
+                0.04039536044001579,
+                0.040337879210710526,
+                0.04028085991740227,
+                0.04022432863712311,
+                0.04016827046871185,
+                0.040112681686878204,
+                0.04005756974220276,
+                0.04000292718410492,
+                0.03994873911142349,
+                0.03989502415060997,
+                0.039841774851083755,
+                0.03978898376226425,
+                0.03973665088415146,
+                0.03968477621674538,
+                0.03963335603475571,
+                0.03958239033818245,
+                0.03953187167644501,
+                0.03948180377483368,
+                0.03943217918276787,
+                0.039382997900247574,
+                0.0393342599272728,
+                0.03928596153855324,
+                0.0392381027340889,
+                0.03919067233800888,
+                0.03914368897676468,
+                0.039097122848033905,
+                0.03905098885297775,
+                0.039005279541015625,
+                0.038959991186857224,
+                0.03891512751579285,
+                0.038870684802532196,
+                0.038826655596494675,
+                0.038783036172389984,
+                0.03873983398079872,
+                0.03869703412055969,
+                0.038654640316963196,
+                0.03861265629529953,
+                0.0385710671544075,
+                0.038529884070158005,
+                0.03848908841609955,
+                0.03844869136810303,
+                0.038408681750297546,
+                0.038369063287973404,
+                0.038329824805259705,
+                0.038290973752737045,
+                0.03825250267982483,
+                0.038214411586523056,
+                0.03817668557167053,
+                0.03813934698700905,
+                0.03810235857963562,
+                0.03806575387716293,
+                0.0380295105278492,
+                0.03799362853169441,
+                0.037958092987537384,
+                0.0379229299724102,
+                0.03788810968399048,
+                0.037853650748729706,
+                0.03781953081488609,
+                0.03778575733304024,
+                0.03775233030319214,
+                0.0377192385494709,
+                0.03768648952245712,
+                0.03765407204627991,
+                0.037621986120939255,
+                0.037590231746435165,
+                0.037558797746896744,
+                0.037527695298194885,
+                0.037496909499168396,
+                0.037466444075107574,
+                0.03743629530072212,
+                0.03740645572543144,
+                0.037376925349235535,
+                0.0373477078974247,
+                0.037318792194128036,
+                0.03729017451405525,
+                0.03726185858249664,
+                0.03723384067416191,
+                0.03720611333847046,
+                0.037178684026002884,
+                0.03715153783559799,
+                0.03712468594312668,
+                0.03709810599684715,
+                0.03707181289792061,
+                0.03704579547047615,
+                0.03702004998922348,
+                0.036994583904743195,
+                0.0369693823158741,
+                0.03694445639848709,
+                0.036919791251420975,
+                0.03689538314938545,
+                0.03687124326825142,
+                0.03684735298156738,
+                0.03682371973991394,
+                0.03680034354329109,
+                0.03677721694111824,
+                0.03675432875752449,
+                0.03673168644309044,
+                0.036709293723106384,
+                0.03668713942170143,
+                0.03666521608829498,
+                0.03664353862404823,
+                0.03662208095192909,
+                0.036600857973098755,
+                0.03657986596226692,
+                0.036559101194143295,
+                0.03653854876756668,
+                0.03651822358369827,
+                0.036498118191957474,
+                0.03647821769118309,
+                0.03645854443311691,
+                0.03643907606601715,
+                0.036419812589883804,
+                0.03640075400471687,
+                0.036381907761096954,
+                0.036363258957862854,
+                0.03634480759501457,
+                0.036326561123132706,
+                0.03630850464105606,
+                0.036290641874074936,
+                0.036272965371608734,
+                0.03625548258423805,
+                0.036238186061382294,
+                0.03622107580304146,
+                0.03620414808392525,
+                0.036187395453453064,
+                0.0361708328127861,
+                0.036154430359601974,
+                0.03613821044564247,
+                0.0361221581697464,
+                0.036106280982494354,
+                0.03609057143330574,
+                0.03607501834630966,
+                0.03605964034795761,
+                0.036044418811798096,
+                0.03602936491370201,
+                0.03601446375250816,
+                0.03599971905350685,
+                0.035985130816698074,
+                0.035970695316791534,
+                0.035956401377916336,
+                0.03594226762652397,
+                0.035928282886743546,
+                0.03591443970799446,
+                0.03590074181556702,
+                0.03588717803359032,
+                0.035873766988515854,
+                0.03586048632860184,
+                0.035847339779138565,
+                0.03583433851599693,
+                0.03582146018743515,
+                0.03580871596932411,
+                0.03579610586166382,
+                0.035783618688583374,
+                0.035771265625953674,
+                0.03575903922319412,
+                0.03574693202972412,
+                0.03573494404554367,
+                0.03572307899594307,
+                0.03571133688092232,
+                0.03569971024990082,
+                0.03568819910287857,
+                0.03567679971456528,
+                0.035665515810251236,
+                0.035654351115226746,
+                0.035643287003040314,
+                0.035632338374853134,
+                0.03562149032950401,
+                0.03561076149344444,
+                0.03560011833906174,
+                0.035589590668678284,
+                0.03557916730642319,
+                0.03556883707642555,
+                0.03555860370397568,
+                0.03554848954081535,
+                0.035538449883461,
+                0.0355285182595253,
+                0.03551867976784706,
+                0.03550893813371658,
+                0.035499271005392075,
+                0.03548971191048622,
+                0.03548023849725723,
+                0.03547084704041481,
+                0.035461556166410446,
+                0.03545233979821205,
+                0.03544321656227112,
+                0.03543417155742645,
+                0.03542520850896835,
+                0.03541632741689682,
+                0.03540753573179245,
+                0.03539881110191345
+              ],
+              "type": "line"
+            }
+          ],
+          "title": {
+            "text": "Training loss"
+          },
+          "xAxis": {
+            "type": "category"
+          },
+          "yAxis": {
+            "type": "value"
+          }
+        },
+        "error": null,
+        "input_metadata": [
+          {
+            "dataframes": {
+              "df": {
+                "columns": [
+                  "x",
+                  "y"
+                ]
+              },
+              "df_test": {
+                "columns": [
+                  "x",
+                  "y"
+                ]
+              },
+              "df_train": {
+                "columns": [
+                  "x",
+                  "y"
+                ]
+              },
+              "training": {
+                "columns": [
+                  "training_loss"
+                ]
+              }
+            },
+            "other": {
+              "model": {
+                "model": {
+                  "inputs": [
+                    "Input__tensor_1_output"
+                  ],
+                  "loss_inputs": [
+                    "Input__tensor_3_output",
+                    "Output_1_x"
+                  ],
+                  "outputs": [
+                    "Output_1_x"
+                  ],
+                  "trained": true
+                },
+                "type": "model"
+              }
+            },
+            "relations": []
+          }
+        ],
+        "meta": {
+          "inputs": {
+            "bundle": {
+              "name": "bundle",
+              "position": "left",
+              "type": {
+                "type": "<class 'lynxkite_graph_analytics.core.Bundle'>"
+              }
+            }
+          },
+          "name": "View loss",
+          "outputs": {},
+          "params": {},
+          "position": {
+            "x": 968.0,
+            "y": 610.0
+          },
+          "type": "visualization"
+        },
+        "params": {},
+        "status": "done",
+        "title": "View loss"
+      },
+      "dragHandle": ".bg-primary",
+      "height": 417.0,
+      "id": "View loss 1",
+      "position": {
+        "x": 2052.643815443704,
+        "y": 617.6466886696832
+      },
+      "type": "visualization",
+      "width": 721.0
     }
   ]
 }

lynxkite-graph-analytics/src/lynxkite_graph_analytics/lynxkite_ops.py

@@ -389,6 +389,7 @@ def train_model(
         losses.append(loss)
     m.trained = True
     bundle = bundle.copy()
+    bundle.dfs["training"] = pd.DataFrame({"training_loss": losses})
     bundle.other[model_name] = m
     return bundle
 
@@ -431,3 +432,17 @@ def train_test_split(bundle: core.Bundle, *, table_name: str, test_ratio: float
     bundle.dfs[f"{table_name}_train"] = train
     bundle.dfs[f"{table_name}_test"] = test
     return bundle
+
+
+@op("View loss", view="visualization")
+def view_loss(
+    bundle: core.Bundle,
+):
+    loss = bundle.dfs["training"].training_loss.tolist()
+    v = {
+        "title": {"text": "Training loss"},
+        "xAxis": {"type": "category"},
+        "yAxis": {"type": "value"},
+        "series": [{"data": loss, "type": "line"}],
+    }
+    return v