isky/backend-springboot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add new prediction service

Browse Source
This commit is contained in:
heyethereum
2024-08-13 02:34:47 +08:00
parent 53f9acd922
commit c8cfe610a6
45 changed files with 167 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/main/java/com/safeqr/app/qrcode/service/URLVerificationService.java
View File

@@ -4,7 +4,7 @@ import static com.safeqr.app.constants.CommonConstants.*;
import com.safeqr.app.qrcode.entity.URLEntity;
import com.safeqr.app.qrcode.model.URLModel;
import com.safeqr.app.qrcode.repository.URLRepository;
import com.safeqr.app.spark.service.MLModelService;
import com.safeqr.app.prediction.service.PredictionService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
@@ -28,11 +28,11 @@ public class URLVerificationService {
private static final int READ_TIMEOUT_MS = 10000;
private static final Logger logger = LoggerFactory.getLogger(URLVerificationService.class);
private final URLRepository urlRepository;
private final MLModelService mlModelService;
private final PredictionService predictionService;
@Autowired
public URLVerificationService(URLRepository urlRepository, MLModelService mlModelService) {
public URLVerificationService(URLRepository urlRepository, PredictionService predictionService) {
this.urlRepository = urlRepository;
this.mlModelService = mlModelService;
this.predictionService = predictionService;
}
// Regular expression pattern for shortening services
@@ -428,6 +428,6 @@ public class URLVerificationService {
// Get Classification using ML Model
public String getClassification(URLModel urlModel){
return mlModelService.predict(urlModel);
return predictionService.predict(urlModel);
}
}