crossvalida

src/components/ModelBuilder.jsx

@@ -0,0 +1,102 @@
+import React, { useState, useEffect } from 'react';
+import Tree from './Tree';
+import { shuffleAndChunkArray, mergeChunksWithoutChosen } from '../utils/cross-valid';
+import { useLoadingContext } from '../contexts/LoadingContext';
+import { executeAlgorithm } from '../utils/algorithm-executor';
+import ConfusionMatrix from './ConfusionMatrix';
+import CrossValidator from './Crossvalidator/Crossvalidator';
+
+/**
+ *
+ * @param {Object} props
+ * @param {Object} props.builder
+ * @param {Array} props.builder.allAttributes
+ * @param {Array} props.builder.allClasses
+ * @param {string} props.builder.categoryAttr
+ * @param {string} props.builder.entropyThrehold
+ * @param {string} props.builder.minItemsCount
+ * @param {Array} props.builder.trainingSet
+ * @param {Array<string>} props.builder.ignoredAttributes
+ * @param {string} props.builder.algorithm
+ * @param {Boolean} props.headers
+ * @param {Boolean} props.isReady
+ * @param {Number} props.fold
+
+ * @returns
+ */
+function ModelBuilder({ builder, headers, fold = 10 }) {
+  const { isLoading, setIsLoading } = useLoadingContext();
+
+  const [mainDataset, setMainDataset] = useState(builder.trainingSet);
+  const [chunks, setChunks] = useState([]);
+  const [treeModels, setTreeModels] = useState([]);
+
+  useEffect(() => {
+    console.log('MODEl BUIDLER');
+    setIsLoading(true);
+
+    let chunks = shuffleAndChunkArray(builder.trainingSet);
+    setChunks(chunks);
+
+    let buildedModels = [];
+    let tmpBuilder = JSON.parse(JSON.stringify(builder));
+
+    chunks.forEach((x, i) => {
+      tmpBuilder.trainingSet = mergeChunksWithoutChosen(chunks, i);
+      console.log('CHUNK', i);
+      executeAlgorithm(tmpBuilder)
+        .then(value => {
+          // if (terminated) {
+          //   return;
+          // }
+          buildedModels.push(value);
+          setTreeModels(buildedModels);
+        })
+        .catch(e => console.error(e))
+        .finally(() => {
+          // if (!terminated) setIsLoading(false);
+          // terminated = true;
+        });
+    });
+    Promise.A;
+    for (let i = 0; i < fold; i++) {
+      tmpBuilder.trainingSet = mergeChunksWithoutChosen(chunks, i);
+      console.log('CHUNK', i);
+      executeAlgorithm(tmpBuilder)
+        .then(value => {
+          // if (terminated) {
+          //   return;
+          // }
+          buildedModels.push(value);
+          setTreeModels(buildedModels);
+        })
+        .catch(e => console.error(e))
+        .finally(() => {
+          // if (!terminated) setIsLoading(false);
+          // terminated = true;
+        });
+    }
+
+    console.log(chunks);
+    console.log(buildedModels);
+    setIsLoading(false);
+    // return () => {
+    //   if (terminated) {
+    //     return;
+    //   }
+    //   terminated = true;
+    // };
+  }, [builder, setIsLoading]);
+
+  return (
+    <div id="model-builder" className="main">
+      <div>
+        do nowego komponentu to dać
+        <CrossValidator builder={builder} chunks={chunks} treeModels={treeModels} />
+        {/* <Tree options={builder} headers={headers} /> */}
+      </div>
+    </div>
+  );
+}
+
+export default ModelBuilder;