| 134 | } |
| 135 | |
| 136 | TDataProviderPtr GetMultiClassPool() { |
| 137 | TSrcData srcData; |
| 138 | srcData.DatasetFileData = // Cloudness pool first 9 features |
| 139 | "0\t0\t1481112000\t1\t52.3699989319\t143.179992676\t1481112000\t41924.0\t0.545332845052\t1\n" |
| 140 | "1\t0\t1484784000\t1\t52.0999984741\t102.699996948\t1484784000\t41777.0\t9.84666646322\t1\n" |
| 141 | "2\t0\t1479567600\t1\t59.0166702271\t54.6500015259\t1479567600\t41493.0\t21.6433334351\t1\n" |
| 142 | "2\t0\t1486112400\t1\t53.3499984741\t75.4499969482\t1486112400\t40746.0\t17.0299997965\t1\n" |
| 143 | "2\t0\t1479772800\t1\t53.8824996948\t28.0307006836\t1479772800\t33919.0\t4.86871337891\t0\n" |
| 144 | "0\t0\t1481760000\t1\t48.2200012207\t46.7299995422\t1481760000\t49599.0\t6.11533330282\t0\n" |
| 145 | "2\t0\t1481878800\t1\t59.6500015259\t154.270004272\t1481878800\t41483.0\t22.2846669515\t0\n" |
| 146 | "1\t0\t1482742800\t1\t54.2999992371\t155.916671753\t1482742800\t41947.0\t22.3944447835\t0\n" |
| 147 | "2\t0\t1480431600\t1\t48.0666694641\t46.1166687012\t1480431600\t49598.0\t21.0744445801\t0\n" |
| 148 | "1\t0\t1484872200\t1\t56.3802986145\t85.2082977295\t1484872200\t33961.0\t8.68055318197\t0\n" |
| 149 | "2\t0\t1486090800\t1\t51.8699989319\t58.1800003052\t1486090800\t41961.0\t9.87866668701\t0\n" |
| 150 | "2\t0\t1485896400\t1\t58.5200004578\t58.8499984741\t1485896400\t41495.0\t3.92333323161\t0\n" |
| 151 | "0\t0\t1485653400\t1\t46.8886985779\t142.718002319\t1485653400\t33779.0\t13.514533488\t1\n" |
| 152 | "2\t0\t1482323400\t1\t55.9725990295\t37.4146003723\t1482323400\t34193.0\t17.4943066915\t0\n" |
| 153 | "1\t0\t1485066600\t1\t51.0222015381\t71.4669036865\t1485066600\t33632.0\t13.7644602458\t1\n" |
| 154 | "0\t0\t1480644000\t1\t55.3058013916\t61.5032997131\t1480644000\t34003.0\t9.10021998088\t0"; |
| 155 | |
| 156 | TStringBuilder cdOutput; |
| 157 | cdOutput << "0\tTarget\n" << "9\tCateg" << Endl; |
| 158 | |
| 159 | TReadDatasetMainParams readDatasetMainParams; |
| 160 | |
| 161 | TVector<THolder<TTempFile>> srcDataFiles; |
| 162 | SaveSrcData(srcData, &readDatasetMainParams, &srcDataFiles); |
| 163 | TVector<NJson::TJsonValue> classLabels; |
| 164 | |
| 165 | return ReadDataset( |
| 166 | /*taskType*/Nothing(), |
| 167 | readDatasetMainParams.PoolPath, |
| 168 | TPathWithScheme(), |
| 169 | TPathWithScheme(), |
| 170 | TPathWithScheme(), |
| 171 | TPathWithScheme(), |
| 172 | TPathWithScheme(), |
| 173 | TPathWithScheme(), |
| 174 | TPathWithScheme(), |
| 175 | readDatasetMainParams.ColumnarPoolFormatParams, |
| 176 | /*ignoredFeatures*/ {}, |
| 177 | EObjectsOrder::Undefined, |
| 178 | /*threadCount*/ 16, |
| 179 | /*verbose*/true, |
| 180 | /*loadSampleIds*/ false, |
| 181 | /*forceUnitAutoPairWeights*/ false, |
| 182 | &classLabels |
| 183 | ); |
| 184 | } |
| 185 | |
| 186 | TFullModel SimpleFloatModel(size_t treeCount) { |
| 187 | TFullModel model; |
no test coverage detected