Files
HowToSD-ComfyUI-Pt-Wrapper/examples/workflows/iris_classification_train_eval.json

1 line
14 KiB
JSON

{"last_node_id":76,"last_link_id":112,"nodes":[{"id":70,"type":"PandasShowDataFrame","pos":[522.3012084960938,4590.89990234375],"size":[497.2306213378906,448.692138671875],"flags":{},"order":7,"mode":0,"inputs":[{"name":"dataframe","type":"DATAFRAME","link":104}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"properties":{"Node name for S&R":"PandasShowDataFrame"},"widgets_values":[" sepal length sepal width petal length petal width class\n0 5.1 3.5 1.4 0.2 Iris-setosa\n1 5.4 3.9 1.7 0.4 Iris-setosa\n2 5.4 3.7 1.5 0.2 Iris-setosa\n3 5.7 4.4 1.5 0.4 Iris-setosa\n4 5.4 3.4 1.7 0.2 Iris-setosa\n5 5.0 3.0 1.6 0.2 Iris-setosa\n6 4.8 3.1 1.6 0.2 Iris-setosa\n7 5.0 3.2 1.2 0.2 Iris-setosa\n8 5.0 3.5 1.3 0.3 Iris-setosa\n9 4.8 3.0 1.4 0.3 Iris-setosa\n10 7.0 3.2 4.7 1.4 Iris-versicolor\n11 5.7 2.8 4.5 1.3 Iris-versicolor\n12 5.0 2.0 3.5 1.0 Iris-versicolor\n13 6.7 3.1 4.4 1.4 Iris-versicolor\n14 5.9 3.2 4.8 1.8 Iris-versicolor\n15 6.6 3.0 4.4 1.4 Iris-versicolor\n16 5.5 2.4 3.8 1.1 Iris-versicolor\n17 6.0 3.4 4.5 1.6 Iris-versicolor\n18 5.5 2.6 4.4 1.2 Iris-versicolor\n19 5.7 3.0 4.2 1.2 Iris-versicolor\n20 6.3 3.3 6.0 2.5 Iris-virginica\n21 7.6 3.0 6.6 2.1 Iris-virginica\n22 6.5 3.2 5.1 2.0 Iris-virginica\n23 6.4 3.2 5.3 2.3 Iris-virginica\n24 6.9 3.2 5.7 2.3 Iris-virginica\n25 7.2 3.2 6.0 1.8 Iris-virginica\n26 7.4 2.8 6.1 1.9 Iris-virginica\n27 7.7 3.0 6.1 2.3 Iris-virginica\n28 6.7 3.1 5.6 2.4 Iris-virginica\n29 6.7 3.0 5.2 2.3 Iris-virginica"]},{"id":68,"type":"PandasReplaceAdvanced","pos":[1072.482666015625,4348.2099609375],"size":[261.212158203125,119.18181610107422],"flags":{},"order":6,"mode":0,"inputs":[{"name":"dataframe","type":"DATAFRAME","link":102}],"outputs":[{"name":"DATAFRAME","type":"DATAFRAME","links":[103],"slot_index":0}],"properties":{"Node name for S&R":"PandasReplaceAdvanced"},"widgets_values":["{\"Iris-setosa\":0,\n\"Iris-versicolor\":1,\n\"Iris-virginica\":2}"]},{"id":44,"type":"CDAShowFloat","pos":[3463.3115234375,4348.2099609375],"size":[315,76],"flags":{},"order":18,"mode":0,"inputs":[{"name":"float_scalar","type":"FLOAT","link":53,"widget":{"name":"float_scalar"}}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"title":"Accuracy","properties":{"Node name for S&R":"CDAShowFloat"},"widgets_values":[0,"0.9666666666666667"]},{"id":72,"type":"CDAShowFloat","pos":[3463.3115234375,4476.33642578125],"size":[315,76],"flags":{},"order":19,"mode":0,"inputs":[{"name":"float_scalar","type":"FLOAT","link":108,"widget":{"name":"float_scalar"}}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"title":"Precision","properties":{"Node name for S&R":"CDAShowFloat"},"widgets_values":[0,"0.9696969696969697"]},{"id":73,"type":"CDAShowFloat","pos":[3463.3115234375,4601.6923828125],"size":[315,76],"flags":{},"order":20,"mode":0,"inputs":[{"name":"float_scalar","type":"FLOAT","link":110,"widget":{"name":"float_scalar"}}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"title":"Recall","properties":{"Node name for S&R":"CDAShowFloat"},"widgets_values":[0,"0.9666666666666667"]},{"id":74,"type":"CDAShowFloat","pos":[3463.3115234375,4735.68310546875],"size":[315,76],"flags":{},"order":21,"mode":0,"inputs":[{"name":"float_scalar","type":"FLOAT","link":111,"widget":{"name":"float_scalar"}}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"title":"F1","properties":{"Node name for S&R":"CDAShowFloat"},"widgets_values":[0,"0.9665831244778612"]},{"id":76,"type":"CDAShowInt","pos":[3463.3115234375,4872.4619140625],"size":[315,76],"flags":{},"order":22,"mode":0,"inputs":[{"name":"int_scalar","type":"INT","link":112,"widget":{"name":"int_scalar"}}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"title":"Num samples","properties":{"Node name for S&R":"CDAShowInt"},"widgets_values":[0,"30"]},{"id":16,"type":"PandasShowDataFrame","pos":[1072.482666015625,3408.90966796875],"size":[481.2576599121094,230.6519775390625],"flags":{},"order":8,"mode":0,"inputs":[{"name":"dataframe","type":"DATAFRAME","link":77}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"properties":{"Node name for S&R":"PandasShowDataFrame"},"widgets_values":[" sepal length sepal width petal length petal width class\n0 4.9 3.0 1.4 0.2 0\n1 4.7 3.2 1.3 0.2 0\n2 4.6 3.1 1.5 0.2 0\n3 5.0 3.6 1.4 0.2 0\n4 4.6 3.4 1.4 0.3 0\n.. ... ... ... ... ...\n115 6.7 3.3 5.7 2.5 2\n116 6.3 2.5 5.0 1.9 2\n117 6.5 3.0 5.2 2.0 2\n118 6.2 3.4 5.4 2.3 2\n119 5.9 3.0 5.1 1.8 2\n\n[120 rows x 5 columns]"]},{"id":11,"type":"PandasShowDataFrame","pos":[522.3012084960938,3408.90966796875],"size":[514.46728515625,237.3878936767578],"flags":{},"order":4,"mode":0,"inputs":[{"name":"dataframe","type":"DATAFRAME","link":88}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"properties":{"Node name for S&R":"PandasShowDataFrame"},"widgets_values":[" sepal length sepal width petal length petal width class\n0 4.9 3.0 1.4 0.2 Iris-setosa\n1 4.7 3.2 1.3 0.2 Iris-setosa\n2 4.6 3.1 1.5 0.2 Iris-setosa\n3 5.0 3.6 1.4 0.2 Iris-setosa\n4 4.6 3.4 1.4 0.3 Iris-setosa\n.. ... ... ... ... ...\n115 6.7 3.3 5.7 2.5 Iris-virginica\n116 6.3 2.5 5.0 1.9 Iris-virginica\n117 6.5 3.0 5.2 2.0 Iris-virginica\n118 6.2 3.4 5.4 2.3 Iris-virginica\n119 5.9 3.0 5.1 1.8 Iris-virginica\n\n[120 rows x 5 columns]"]},{"id":63,"type":"PtDataLoaderFromTensors","pos":[2156.013916015625,4348.2099609375],"size":[340.20001220703125,102],"flags":{},"order":12,"mode":0,"inputs":[{"name":"x","type":"TENSOR","link":106},{"name":"y","type":"TENSOR","link":107}],"outputs":[{"name":"PTDATALOADER","type":"PTDATALOADER","links":[75],"slot_index":0}],"properties":{"Node name for S&R":"PtDataLoaderFromTensors"},"widgets_values":[5,false]},{"id":29,"type":"PtnLinearModel","pos":[2156.013916015625,3441.72314453125],"size":[315,106],"flags":{},"order":0,"mode":0,"inputs":[],"outputs":[{"name":"PTMODEL","type":"PTMODEL","links":[29,51],"slot_index":0}],"properties":{"Node name for S&R":"PtnLinearModel"},"widgets_values":["[4,512,3]","[True,True]",2]},{"id":42,"type":"PtoAdam","pos":[2156.013916015625,3696.825927734375],"size":[315,106],"flags":{},"order":3,"mode":0,"inputs":[{"name":"model","type":"PTMODEL","link":51}],"outputs":[{"name":"PTOPTIMIZER","type":"PTOPTIMIZER","links":[52],"slot_index":0}],"properties":{"Node name for S&R":"PtoAdam"},"widgets_values":[0.001,0.9,0.999]},{"id":32,"type":"PtSaveModel","pos":[3463.3115234375,3173.329833984375],"size":[315,58],"flags":{},"order":14,"mode":0,"inputs":[{"name":"model","type":"PTMODEL","link":32}],"outputs":[{"name":"model full path","type":"STRING","links":[49],"slot_index":0}],"properties":{"Node name for S&R":"PtSaveModel"},"widgets_values":["iris.pkl"]},{"id":33,"type":"PtShowText","pos":[3463.3115234375,3343.487548828125],"size":[631.916259765625,81.7053451538086],"flags":{},"order":16,"mode":0,"inputs":[{"name":"tens","type":"TENSOR","link":33}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"properties":{"Node name for S&R":"PtShowText"},"widgets_values":["tensor([0.9147, 0.6376, 0.4849, ..., 0.0641, 0.0745, 0.0556])"]},{"id":65,"type":"PandasLoadCSVWithIndex","pos":[522.3012084960938,3173.329833984375],"size":[327.5999755859375,82],"flags":{},"order":1,"mode":0,"inputs":[],"outputs":[{"name":"DATAFRAME","type":"DATAFRAME","links":[88,89],"slot_index":0}],"properties":{"Node name for S&R":"PandasLoadCSVWithIndex"},"widgets_values":["custom_nodes/pt-wrapper/examples/datasets/iris/iris_train.csv",0]},{"id":69,"type":"PandasFeatureSplitToPt","pos":[1702.9251708984375,4348.2099609375],"size":[327.5999755859375,150],"flags":{},"order":10,"mode":0,"inputs":[{"name":"dataframe","type":"DATAFRAME","link":103}],"outputs":[{"name":"feature","type":"TENSOR","links":[106],"slot_index":0},{"name":"label","type":"TENSOR","links":[107],"slot_index":1}],"properties":{"Node name for S&R":"PandasFeatureSplitToPt"},"widgets_values":[-1,true,"float32","int64"]},{"id":64,"type":"PandasReplaceAdvanced","pos":[1072.482666015625,3173.329833984375],"size":[261.212158203125,119.18181610107422],"flags":{},"order":5,"mode":0,"inputs":[{"name":"dataframe","type":"DATAFRAME","link":89}],"outputs":[{"name":"DATAFRAME","type":"DATAFRAME","links":[77,90],"slot_index":0}],"properties":{"Node name for S&R":"PandasReplaceAdvanced"},"widgets_values":["{\"Iris-setosa\":0,\n\"Iris-versicolor\":1,\n\"Iris-virginica\":2}"]},{"id":66,"type":"PandasFeatureSplitToPt","pos":[1702.9251708984375,3173.329833984375],"size":[327.5999755859375,150],"flags":{},"order":9,"mode":0,"inputs":[{"name":"dataframe","type":"DATAFRAME","link":90}],"outputs":[{"name":"feature","type":"TENSOR","links":[91],"slot_index":0},{"name":"label","type":"TENSOR","links":[92],"slot_index":1}],"properties":{"Node name for S&R":"PandasFeatureSplitToPt"},"widgets_values":[-1,true,"float32","int64"]},{"id":28,"type":"PtDataLoaderFromTensors","pos":[2156.013916015625,3173.329833984375],"size":[340.20001220703125,102],"flags":{},"order":11,"mode":0,"inputs":[{"name":"x","type":"TENSOR","link":91},{"name":"y","type":"TENSOR","link":92}],"outputs":[{"name":"PTDATALOADER","type":"PTDATALOADER","links":[30],"slot_index":0}],"properties":{"Node name for S&R":"PtDataLoaderFromTensors"},"widgets_values":[10,true]},{"id":43,"type":"PtEvaluateClassificationModel","pos":[2724.5361328125,4348.2099609375],"size":[403.1999816894531,138],"flags":{},"order":15,"mode":0,"inputs":[{"name":"model","type":"PTMODEL","link":54},{"name":"data_loader","type":"PTDATALOADER","link":75}],"outputs":[{"name":"accuracy","type":"FLOAT","links":[53],"slot_index":0},{"name":"precision","type":"FLOAT","links":[108],"slot_index":1},{"name":"recall","type":"FLOAT","links":[110],"slot_index":2},{"name":"f1","type":"FLOAT","links":[111],"slot_index":3},{"name":"num_samples","type":"INT","links":[112],"slot_index":4}],"properties":{"Node name for S&R":"PtEvaluateClassificationModel"},"widgets_values":[true]},{"id":30,"type":"PtTrainClassificationModel","pos":[2724.5361328125,3173.329833984375],"size":[365.4000244140625,214],"flags":{},"order":13,"mode":0,"inputs":[{"name":"model","type":"PTMODEL","link":29},{"name":"train_loader","type":"PTDATALOADER","link":30},{"name":"optimizer","type":"PTOPTIMIZER","link":52},{"name":"val_loader","type":"PTDATALOADER","link":null,"shape":7}],"outputs":[{"name":"Model","type":"PTMODEL","links":[32,54],"slot_index":0},{"name":"train loss","type":"TENSOR","links":[33],"slot_index":1},{"name":"val loss","type":"TENSOR","links":null}],"properties":{"Node name for S&R":"PtTrainClassificationModel"},"widgets_values":[200,true,false,10,false]},{"id":41,"type":"PandasShowText","pos":[4206.8994140625,3173.329833984375],"size":[523.6947021484375,76],"flags":{},"order":17,"mode":0,"inputs":[{"name":"text","type":"STRING","link":49,"widget":{"name":"text"}}],"outputs":[{"name":"STRING","type":"STRING","links":null,"shape":6}],"properties":{"Node name for S&R":"PandasShowText"},"widgets_values":["","/home/dockeruser/comfy/ComfyUI/custom_nodes/pt-wrapper/models/iris.pkl"]},{"id":67,"type":"PandasLoadCSVWithIndex","pos":[522.3012084960938,4348.2099609375],"size":[327.5999755859375,82],"flags":{},"order":2,"mode":0,"inputs":[],"outputs":[{"name":"DATAFRAME","type":"DATAFRAME","links":[102,104],"slot_index":0}],"properties":{"Node name for S&R":"PandasLoadCSVWithIndex"},"widgets_values":["custom_nodes/pt-wrapper/examples/datasets/iris/iris_val.csv",0]}],"links":[[29,29,0,30,0,"PTMODEL"],[30,28,0,30,1,"PTDATALOADER"],[32,30,0,32,0,"PTMODEL"],[33,30,1,33,0,"TENSOR"],[49,32,0,41,0,"STRING"],[51,29,0,42,0,"PTMODEL"],[52,42,0,30,2,"PTOPTIMIZER"],[53,43,0,44,0,"FLOAT"],[54,30,0,43,0,"PTMODEL"],[75,63,0,43,1,"PTDATALOADER"],[77,64,0,16,0,"DATAFRAME"],[88,65,0,11,0,"DATAFRAME"],[89,65,0,64,0,"DATAFRAME"],[90,64,0,66,0,"DATAFRAME"],[91,66,0,28,0,"TENSOR"],[92,66,1,28,1,"TENSOR"],[102,67,0,68,0,"DATAFRAME"],[103,68,0,69,0,"DATAFRAME"],[104,67,0,70,0,"DATAFRAME"],[106,69,0,63,0,"TENSOR"],[107,69,1,63,1,"TENSOR"],[108,43,1,72,0,"FLOAT"],[110,43,2,73,0,"FLOAT"],[111,43,3,74,0,"FLOAT"],[112,43,4,76,0,"INT"]],"groups":[],"config":{},"extra":{"ds":{"scale":0.2547179056219536,"offset":[-1778.4133821565692,-2306.537758548792]},"node_versions":{"data-analysis":"e490b620286600930c8c9d3a6986f73d5320463f","pt-wrapper":"4fbfd28f609278974f0f2e7cbb8b7bdc4fafef3d"},"ue_links":[]},"version":0.4}