15 changed files with 163 additions and 225 deletions
@ -1,46 +1,46 @@ |
|||||||
package com.mh.algorithm.knn; |
//package com.mh.algorithm.knn;
|
||||||
|
//
|
||||||
import com.mh.algorithm.matrix.Matrix; |
//import com.mh.algorithm.matrix.Matrix;
|
||||||
import com.mh.algorithm.utils.CsvInfo; |
//import com.mh.algorithm.utils.CsvInfo;
|
||||||
import com.mh.algorithm.utils.CsvUtil; |
//import com.mh.algorithm.utils.CsvUtil;
|
||||||
import com.mh.algorithm.utils.DoubleUtil; |
//import com.mh.algorithm.utils.DoubleUtil;
|
||||||
import org.junit.Test; |
//import org.junit.Test;
|
||||||
|
//
|
||||||
/** |
///**
|
||||||
* @program: top-algorithm-set |
// * @program: top-algorithm-set
|
||||||
* @description: |
// * @description:
|
||||||
* @author: Mr.Zhao |
// * @author: Mr.Zhao
|
||||||
* @create: 2020-10-26 22:04 |
// * @create: 2020-10-26 22:04
|
||||||
**/ |
// **/
|
||||||
public class knnTest { |
//public class knnTest {
|
||||||
@Test |
// @Test
|
||||||
public void test() throws Exception { |
// public void test() throws Exception {
|
||||||
// 训练集
|
// // 训练集
|
||||||
CsvInfo csvInfo = CsvUtil.getCsvInfo(false, "E:\\jarTest\\trainData.csv"); |
// CsvInfo csvInfo = CsvUtil.getCsvInfo(false, "E:\\jarTest\\trainData.csv");
|
||||||
Matrix trainSet = csvInfo.toMatrix(); |
// Matrix trainSet = csvInfo.toMatrix();
|
||||||
Matrix trainSetLabels = trainSet.getColOfIdx(trainSet.getMatrixColCount() - 1); |
// Matrix trainSetLabels = trainSet.getColOfIdx(trainSet.getMatrixColCount() - 1);
|
||||||
Matrix trainSetData = trainSet.subMatrix(0, trainSet.getMatrixRowCount(), 0, trainSet.getMatrixColCount() - 1); |
// Matrix trainSetData = trainSet.subMatrix(0, trainSet.getMatrixRowCount(), 0, trainSet.getMatrixColCount() - 1);
|
||||||
|
//
|
||||||
CsvInfo csvInfo1 = CsvUtil.getCsvInfo(false, "E:\\jarTest\\testData.csv"); |
// CsvInfo csvInfo1 = CsvUtil.getCsvInfo(false, "E:\\jarTest\\testData.csv");
|
||||||
Matrix testSet = csvInfo1.toMatrix(); |
// Matrix testSet = csvInfo1.toMatrix();
|
||||||
Matrix testSetData = trainSet.subMatrix(0, testSet.getMatrixRowCount(), 0, testSet.getMatrixColCount() - 1); |
// Matrix testSetData = trainSet.subMatrix(0, testSet.getMatrixRowCount(), 0, testSet.getMatrixColCount() - 1);
|
||||||
Matrix testSetLabels = trainSet.getColOfIdx(testSet.getMatrixColCount() - 1); |
// Matrix testSetLabels = trainSet.getColOfIdx(testSet.getMatrixColCount() - 1);
|
||||||
|
//
|
||||||
// 分类
|
// // 分类
|
||||||
long startTime = System.currentTimeMillis(); |
// long startTime = System.currentTimeMillis();
|
||||||
Matrix result = KNN.classify(testSetData, trainSetData, trainSetLabels, 5); |
// Matrix result = KNN.classify(testSetData, trainSetData, trainSetLabels, 5);
|
||||||
long endTime = System.currentTimeMillis(); |
// long endTime = System.currentTimeMillis();
|
||||||
System.out.println("run time:" + (endTime - startTime)); |
// System.out.println("run time:" + (endTime - startTime));
|
||||||
// 正确率
|
// // 正确率
|
||||||
Matrix error = result.subtract(testSetLabels); |
// Matrix error = result.subtract(testSetLabels);
|
||||||
int total = error.getMatrixRowCount(); |
// int total = error.getMatrixRowCount();
|
||||||
int correct = 0; |
// int correct = 0;
|
||||||
for (int i = 0; i < error.getMatrixRowCount(); i++) { |
// for (int i = 0; i < error.getMatrixRowCount(); i++) {
|
||||||
if (DoubleUtil.equals(error.getValOfIdx(i, 0), 0.0)) { |
// if (DoubleUtil.equals(error.getValOfIdx(i, 0), 0.0)) {
|
||||||
correct++; |
// correct++;
|
||||||
} |
// }
|
||||||
} |
// }
|
||||||
double correctRate = Double.valueOf(correct) / Double.valueOf(total); |
// double correctRate = Double.valueOf(correct) / Double.valueOf(total);
|
||||||
System.out.println("correctRate:"+ correctRate); |
// System.out.println("correctRate:"+ correctRate);
|
||||||
} |
// }
|
||||||
} |
//}
|
||||||
|
|||||||
@ -1,13 +1,13 @@ |
|||||||
package com.mh.common.annotation; |
//package com.mh.common.annotation;
|
||||||
|
//
|
||||||
import java.lang.annotation.*; |
//import java.lang.annotation.*;
|
||||||
|
//
|
||||||
/** |
///**
|
||||||
* Created by fangzhipeng on 2017/7/12. |
// * Created by fangzhipeng on 2017/7/12.
|
||||||
*/ |
// */
|
||||||
@Target(ElementType.METHOD) |
//@Target(ElementType.METHOD)
|
||||||
@Retention(RetentionPolicy.RUNTIME) |
//@Retention(RetentionPolicy.RUNTIME)
|
||||||
@Documented |
//@Documented
|
||||||
public @interface SysLogger { |
//public @interface SysLogger {
|
||||||
String value() default ""; |
// String value() default "";
|
||||||
} |
//}
|
||||||
|
|||||||
@ -1,42 +1,42 @@ |
|||||||
package com.mh.common.utils; |
//package com.mh.common.utils;
|
||||||
|
//
|
||||||
import javax.servlet.http.HttpServletResponse; |
//import javax.servlet.http.HttpServletResponse;
|
||||||
import java.io.BufferedInputStream; |
//import java.io.BufferedInputStream;
|
||||||
import java.io.BufferedOutputStream; |
//import java.io.BufferedOutputStream;
|
||||||
import java.io.File; |
//import java.io.File;
|
||||||
import java.io.FileInputStream; |
//import java.io.FileInputStream;
|
||||||
import java.io.InputStream; |
//import java.io.InputStream;
|
||||||
|
//
|
||||||
/** |
///**
|
||||||
* 文件相关操作 |
// * 文件相关操作
|
||||||
* @author Louis |
// * @author Louis
|
||||||
* @date Jan 14, 2019 |
// * @date Jan 14, 2019
|
||||||
*/ |
// */
|
||||||
public class FileUtils { |
//public class FileUtils {
|
||||||
|
//
|
||||||
/** |
// /**
|
||||||
* 下载文件 |
// * 下载文件
|
||||||
* @param response |
// * @param response
|
||||||
* @param file |
// * @param file
|
||||||
* @param newFileName |
// * @param newFileName
|
||||||
*/ |
// */
|
||||||
public static void downloadFile(HttpServletResponse response, File file, String newFileName) { |
// public static void downloadFile(HttpServletResponse response, File file, String newFileName) {
|
||||||
try { |
// try {
|
||||||
response.setHeader("Content-Disposition", "attachment; filename=" + new String(newFileName.getBytes("ISO-8859-1"), "UTF-8")); |
// response.setHeader("Content-Disposition", "attachment; filename=" + new String(newFileName.getBytes("ISO-8859-1"), "UTF-8"));
|
||||||
BufferedOutputStream bos = new BufferedOutputStream(response.getOutputStream()); |
// BufferedOutputStream bos = new BufferedOutputStream(response.getOutputStream());
|
||||||
InputStream is = new FileInputStream(file.getAbsolutePath()); |
// InputStream is = new FileInputStream(file.getAbsolutePath());
|
||||||
BufferedInputStream bis = new BufferedInputStream(is); |
// BufferedInputStream bis = new BufferedInputStream(is);
|
||||||
int length = 0; |
// int length = 0;
|
||||||
byte[] temp = new byte[1 * 1024 * 10]; |
// byte[] temp = new byte[1 * 1024 * 10];
|
||||||
while ((length = bis.read(temp)) != -1) { |
// while ((length = bis.read(temp)) != -1) {
|
||||||
bos.write(temp, 0, length); |
// bos.write(temp, 0, length);
|
||||||
} |
// }
|
||||||
bos.flush(); |
// bos.flush();
|
||||||
bis.close(); |
// bis.close();
|
||||||
bos.close(); |
// bos.close();
|
||||||
is.close(); |
// is.close();
|
||||||
} catch (Exception e) { |
// } catch (Exception e) {
|
||||||
e.printStackTrace(); |
// e.printStackTrace();
|
||||||
} |
// }
|
||||||
} |
// }
|
||||||
} |
//}
|
||||||
|
|||||||
Loading…
Reference in new issue