masskit_ai.spectrum.peptide.models package

Submodules

masskit_ai.spectrum.peptide.models.AIomicsRNN module

class masskit_ai.spectrum.peptide.models.AIomicsRNN.AIomicsRNN(*args: Any, **kwargs: Any)

Bases: SpectrumModel

forward(x)
total_params()

masskit_ai.spectrum.peptide.models.aiomic module

class masskit_ai.spectrum.peptide.models.aiomic.AIomicsModel(*args: Any, **kwargs: Any)

Bases: SpectrumModel

forward(inp)
total_params()
class masskit_ai.spectrum.peptide.models.aiomic.AIomicsModelUnc(*args: Any, **kwargs: Any)

Bases: SpectrumModel

forward(inp)
total_params()
class masskit_ai.spectrum.peptide.models.aiomic.CompoundConv(*args: Any, **kwargs: Any)

Bases: SpectrumModule

compound convolution

forward(inp)

forward pass of compound convolution module

Parameters:

inp – input tensor

Returns:

output tensor

class masskit_ai.spectrum.peptide.models.aiomic.PFCompoundConv(*args: Any, **kwargs: Any)

Bases: Module

forward(inp)
class masskit_ai.spectrum.peptide.models.aiomic.PredFull(*args: Any, **kwargs: Any)

Bases: SpectrumModel

forward(inp)
total_params()
class masskit_ai.spectrum.peptide.models.aiomic.ResBlock(*args: Any, **kwargs: Any)

Bases: SpectrumModule

residual block

forward(inp)

forward pass for residual module

Parameters:

inp – input tensor

Returns:

the block

class masskit_ai.spectrum.peptide.models.aiomic.SqEx(*args: Any, **kwargs: Any)

Bases: SpectrumModule

forward(inp)
class masskit_ai.spectrum.peptide.models.aiomic.pfblock(*args: Any, **kwargs: Any)

Bases: SpectrumModule

forward(inp)
masskit_ai.spectrum.peptide.models.aiomic.test_modifier(model, config)

test version of a modifing function used to modify a model during transfer learning

Parameters:

  • model – the model

  • config – configuration

Returns:

modified model

masskit_ai.spectrum.peptide.models.dense module

class masskit_ai.spectrum.peptide.models.dense.DenseSpectrumNet(*args: Any, **kwargs: Any)

Bases: SpectrumModel

simple n layer dense network

forward(x)
class masskit_ai.spectrum.peptide.models.dense.XorModel(*args: Any, **kwargs: Any)

Bases: SpectrumModel

forward(x)

masskit_ai.spectrum.peptide.models.flipyflopy module

class masskit_ai.spectrum.peptide.models.flipyflopy.FFN(*args: Any, **kwargs: Any)

Bases: SpectrumModule

forward(inp)
class masskit_ai.spectrum.peptide.models.flipyflopy.FlipyFlopy(*args: Any, **kwargs: Any)

Bases: SpectrumModel

forward(inp)
masskit_ai.spectrum.peptide.models.flipyflopy.Init(P)
class masskit_ai.spectrum.peptide.models.flipyflopy.TalkingHeads(*args: Any, **kwargs: Any)

Bases: SpectrumModule

forward(inp)
class masskit_ai.spectrum.peptide.models.flipyflopy.TransBlock(*args: Any, **kwargs: Any)

Bases: SpectrumModule

forward(inp)

Module contents