net
.
sansa_stack
.
ml
.
spark
.
kge
.
linkprediction
.
models
TransE
Related Doc:
package models
class
TransE
extends
Models
Linear Supertypes
Models
,
AnyRef
,
Any
Ordering
Alphabetic
By Inheritance
Inherited
TransE
Models
AnyRef
Any
Hide All
Show All
Visibility
Public
All
Instance Constructors
new
TransE
(
train:
Dataset
[
IntegerTriples
]
,
ne:
Int
,
nr:
Int
,
batch:
Int
,
k:
Int
,
margin:
Float
,
L:
String
,
sk:
SparkSession
)
Value Members
final
def
!=
(
arg0:
Any
)
:
Boolean
Definition Classes
AnyRef → Any
final
def
##
()
:
Int
Definition Classes
AnyRef → Any
final
def
==
(
arg0:
Any
)
:
Boolean
Definition Classes
AnyRef → Any
def
L1
(
vec:
Tensor
[
Float
]
)
:
Float
Definition Classes
Models
def
L2
(
vec:
Tensor
[
Float
]
)
:
Float
Definition Classes
Models
val
Ne
:
Int
Definition Classes
Models
val
Nr
:
Int
Definition Classes
Models
final
def
asInstanceOf
[
T0
]
:
T0
Definition Classes
Any
def
clone
()
:
AnyRef
Attributes
protected[
java.lang
]
Definition Classes
AnyRef
Annotations
@throws
(
...
)
def
dist
(
row:
IntegerTriples
)
:
Tensor
[
Float
]
def
dist
(
data:
Dataset
[
IntegerTriples
]
)
:
Float
var
e
:
Tensor
[
Float
]
Definition Classes
Models
val
epochs
:
Int
final
def
eq
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
def
equals
(
arg0:
Any
)
:
Boolean
Definition Classes
AnyRef → Any
def
finalize
()
:
Unit
Attributes
protected[
java.lang
]
Definition Classes
AnyRef
Annotations
@throws
(
classOf[java.lang.Throwable]
)
final
def
getClass
()
:
Class
[_]
Definition Classes
AnyRef → Any
def
hashCode
()
:
Int
Definition Classes
AnyRef → Any
def
initialize
(
size:
Int
)
:
Tensor
[
Float
]
Definition Classes
Models
final
def
isInstanceOf
[
T0
]
:
Boolean
Definition Classes
Any
val
myL
: (
Tensor
[
Float
]) ⇒
Float
final
def
ne
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
def
negative
(
data:
Dataset
[
IntegerTriples
]
)
:
Dataset
[
IntegerTriples
]
Definition Classes
Models
def
normalize
(
data:
Tensor
[
Float
]
)
:
Tensor
[
Float
]
Definition Classes
Models
final
def
notify
()
:
Unit
Definition Classes
AnyRef
final
def
notifyAll
()
:
Unit
Definition Classes
AnyRef
var
opt
:
Adam
[
Float
]
var
r
:
Tensor
[
Float
]
Definition Classes
Models
val
rate
:
Float
def
run
()
:
Unit
val
seed
:
Random
Definition Classes
Models
def
subset
(
data:
Dataset
[
IntegerTriples
]
)
:
Dataset
[
IntegerTriples
]
Definition Classes
Models
final
def
synchronized
[
T0
]
(
arg0: ⇒
T0
)
:
T0
Definition Classes
AnyRef
def
toString
()
:
String
Definition Classes
AnyRef → Any
def
tuple
(
aux:
IntegerTriples
)
:
IntegerTriples
Definition Classes
Models
final
def
wait
()
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
final
def
wait
(
arg0:
Long
,
arg1:
Int
)
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
final
def
wait
(
arg0:
Long
)
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
Inherited from
Models
Inherited from
AnyRef
Inherited from
Any
Ungrouped