Index

_

__init__() (in module csdata.CSData) (in module csmodel.CSModel) (in module prompt_formatter.PromptFormatter)

__str__() (in module csdata.CSData) (in module csmodel.CSModel)

A

adata_to_arrow() (in module csdata.CSData)

B

benchmark_expression_conversion() (in module utils)

build_arrow_dataset() (in module utils)

C

concat_vocabularies() (in module utils) CSData() (in module csdata)

csdata_from_arrow() (in module csdata.CSData) csdata_from_multiple_arrow_datasets() (in module csdata.CSData) CSModel() (in module csmodel)

E

embed_cell() (in module csmodel.CSModel)

embed_cells() (in module tasks) embed_cells_batched() (in module csmodel.CSModel)

F

fine_tune() (in module csmodel.CSModel)

format_hf_ds() (in module prompt_formatter.PromptFormatter)

G

generate_cells_conditioned_on_cell_type() (in module tasks) generate_from_prompt() (in module csmodel.CSModel) generate_from_prompt_batched() (in module csmodel.CSModel) generate_sentences() (in module utils)

generate_vocabulary() (in module utils) get_benchmark_df() (in module utils) get_cell_sentence_str() (in module prompt_formatter) get_sentence_strings() (in module csdata.CSData)

P

post_process_generated_cell_sentences() (in module utils) predict_cell_types_of_data() (in module tasks)

PromptFormatter() (in module prompt_formatter) push_model_to_hub() (in module csmodel.CSModel)

R

reconstruct_expression_from_cell_sentence() (in module utils)

S

sort_transcript_counts() (in module utils)

T

tokenize_all() (in module utils)

tokenize_loss_on_response() (in module utils) train_test_split_arrow_ds() (in module utils)