Skip to content
This repository has been archived by the owner on Aug 11, 2023. It is now read-only.

Commit

Permalink
Not working ML model
Browse files Browse the repository at this point in the history
  • Loading branch information
RealHinome authored Aug 11, 2023
1 parent a745d1c commit 359dcec
Showing 1 changed file with 3 additions and 7 deletions.
10 changes: 3 additions & 7 deletions helpers/verificator.go
Original file line number Diff line number Diff line change
@@ -1,14 +1,9 @@
package helpers

import (
tf "github.com/galeone/tensorflow/tensorflow/go"
tg "github.com/galeone/tfgo"
)

// CheckHTML uses machine learning to detect fake
// websites such as pishing ones
func CheckHTML(html string) float32 {
model := tg.LoadModel("models/phishing", []string{"serve"}, nil)
/*model := tg.LoadModel("models/phishing", []string{"serve"}, nil)
input, err := tf.NewTensor(html)
if err != nil {
Expand All @@ -21,5 +16,6 @@ func CheckHTML(html string) float32 {
model.Op("serving_default_inputs_input", 0): input,
})
return results[0].Value().(float32)
return results[0].Value().(float32)*/
return 0
}

0 comments on commit 359dcec

Please sign in to comment.