Package tech.molecules.deep
Class MolEncoder
- java.lang.Object
-
- tech.molecules.deep.MolEncoder
-
public class MolEncoder extends Object
Format: Each atom is encoded with the same information: 1. One-Hot Atom Information about bonds to all n other atoms. For pair of atom i and j: if i and j are the same, or if i and j are not connected: all zeros otherwise: 1.
-
-
Constructor Summary
Constructors Constructor Description MolEncoder()
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description BitSetencodeAtom(int zi)BitSetencodeAtomWithConnections(int zi, int[] permutation)BitSetencodeConnection(int zi, int zj)List<BitSet>encodeMolecule()List<BitSet>encodeMolecule(int[] permutation)intgetAtomEncodingLength()intgetConnectionEncodingLengthPerConnection()intgetMoleculeEncodingLength()static voidmain(String[] args)static BitSetmultiHot(int[] set, int length)static BitSetoneHot(int set, int length)static voidprint(List<BitSet> bs, int lengthPerBitSet)voidsetMolecule(com.actelion.research.chem.StereoMolecule m)
-
-
-
Method Detail
-
setMolecule
public void setMolecule(com.actelion.research.chem.StereoMolecule m)
-
getMoleculeEncodingLength
public int getMoleculeEncodingLength()
-
getAtomEncodingLength
public int getAtomEncodingLength()
-
getConnectionEncodingLengthPerConnection
public int getConnectionEncodingLengthPerConnection()
-
encodeAtomWithConnections
public BitSet encodeAtomWithConnections(int zi, int[] permutation)
-
encodeAtom
public BitSet encodeAtom(int zi)
-
encodeConnection
public BitSet encodeConnection(int zi, int zj)
-
oneHot
public static BitSet oneHot(int set, int length)
-
multiHot
public static BitSet multiHot(int[] set, int length)
-
main
public static void main(String[] args)
-
-