Upload folder using huggingface_hub

del_model/README.md

@@ -0,0 +1,88 @@
+---
+library_name: transformers
+tags:
+- generated_from_trainer
+datasets:
+- crispr_data
+model-index:
+- name: SX_ispymac_Lindel_del
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# SX_ispymac_Lindel_del
+
+This model is a fine-tuned version of [](https://huggingface.co/) on the crispr_data dataset.
+It achieves the following results on the evaluation set:
+- Loss: 463.4343
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 0.001
+- train_batch_size: 100
+- eval_batch_size: 100
+- seed: 63036
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- lr_scheduler_warmup_ratio: 0.05
+- num_epochs: 30.0
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| 1638296.8344  | 1.0   | 326  | 1449155.375     |
+| 1164437.5951  | 2.0   | 652  | 871450.75       |
+| 659014.773    | 3.0   | 978  | 472353.125      |
+| 345279.4356   | 4.0   | 1304 | 236237.5781     |
+| 165353.8528   | 5.0   | 1630 | 106501.5078     |
+| 70656.4785    | 6.0   | 1956 | 42171.0586      |
+| 26332.638     | 7.0   | 2282 | 14457.2051      |
+| 8663.8098     | 8.0   | 2608 | 4628.3350       |
+| 2883.4448     | 9.0   | 2934 | 1723.0411       |
+| 1233.9951     | 10.0  | 3260 | 900.9794        |
+| 746.439       | 11.0  | 3586 | 638.7797        |
+| 582.4893      | 12.0  | 3912 | 545.1293        |
+| 518.9121      | 13.0  | 4238 | 505.4234        |
+| 491.481       | 14.0  | 4564 | 487.2142        |
+| 478.4722      | 15.0  | 4890 | 478.1264        |
+| 471.7958      | 16.0  | 5216 | 473.6396        |
+| 467.9093      | 17.0  | 5542 | 470.8682        |
+| 465.5827      | 18.0  | 5868 | 469.4048        |
+| 463.9871      | 19.0  | 6194 | 468.4793        |
+| 462.7912      | 20.0  | 6520 | 467.4167        |
+| 461.7637      | 21.0  | 6846 | 466.8257        |
+| 460.9475      | 22.0  | 7172 | 465.8925        |
+| 460.0868      | 23.0  | 7498 | 465.6684        |
+| 459.2581      | 24.0  | 7824 | 465.3550        |
+| 458.514       | 25.0  | 8150 | 464.5980        |
+| 457.7133      | 26.0  | 8476 | 464.2821        |
+| 456.9633      | 27.0  | 8802 | 464.1039        |
+| 456.1794      | 28.0  | 9128 | 463.6459        |
+| 455.3521      | 29.0  | 9454 | 463.4857        |
+| 454.5061      | 30.0  | 9780 | 463.4343        |
+
+
+### Framework versions
+
+- Transformers 4.44.2
+- Pytorch 2.4.0+cu124
+- Datasets 2.21.0
+- Tokenizers 0.19.1

del_model/config.json

@@ -1,5 +1,4 @@
 {
-  "_name_or_path": "/home/ljw/sdc1/CRISPR_results/Lindel/SX_ispymac_Lindel_del",
   "architectures": [
     "LindelModel"
   ],

del_model/model.py

@@ -0,0 +1,73 @@
+from transformers import PretrainedConfig, PreTrainedModel
+import torch.nn as nn
+import torch
+import torch.nn.functional as F
+
+class LindelConfig(PretrainedConfig):
+    model_type = "Lindel"
+    label_names = ["count"]
+
+    def __init__(
+        self,
+        dlen = 30, # the upper limit of deletion length (strictly less than dlen)
+        mh_len = 4, # the upper limit of micro-homology length
+        model = "indel", # the actual model, should be "indel", "del", or "ins"
+        reg_mode = "l2", # regularization method, should be "l2" or "l1"
+        reg_const = 0.01, # regularization coefficient
+        seed = 63036, # random seed for intialization
+        **kwargs,
+    ):
+        self.dlen = dlen
+        self.mh_len = mh_len
+        self.model = model
+        self.reg_mode = reg_mode
+        self.reg_const = reg_const
+        self.seed = seed
+        super().__init__(**kwargs)
+
+class LindelModel(PreTrainedModel):
+    config_class = LindelConfig
+
+    def __init__(self, config) -> None:
+        super().__init__(config)
+        # In more recent versions of PyTorch, you no longer need to explicitly register_parameter, it's enough to set a member of your nn.Module with nn.Parameter to "notify" pytorch that this variable should be treated as a trainable parameter (https://stackoverflow.com/questions/59234238/how-to-add-parameters-in-module-class-in-pytorch-custom-model).
+        self.generator = torch.Generator().manual_seed(config.seed)
+        self.reg_mode = config.reg_mode
+        self.reg_const = config.reg_const
+        if config.model == "indel":
+            # onehotencoder(ref[cut-17:cut+3])
+            feature_dim = 20 * 4 + 19 * 16
+            class_dim = 2
+        elif config.model == "ins":
+            # onehotencoder(ref[cut-3:cut+3])
+            feature_dim = 6 * 4 + 5 * 16
+            class_dim = 21
+        elif config.model == "del":
+            class_dim = (4 + 1 + 4 + config.dlen - 1) * (config.dlen - 1) // 2
+            # concatenate get_feature and onehotencoder(ref[cut-17:cut+3])
+            feature_dim = class_dim * (config.mh_len + 1) + 20 * 4 + 19 * 16
+        self.linear = nn.Linear(in_features=feature_dim, out_features=class_dim)
+        self.initialize_weights()
+
+    def initialize_weights(self):
+        for m in self.modules():
+            if isinstance(m, nn.Linear):
+                nn.init.normal_(m.weight, mean=0, std=1, generator=self.generator)
+                if m.bias is not None:
+                    nn.init.constant_(m.bias, 0)
+
+    def forward(self, input, count=None) -> torch.Tensor:
+        logit = self.linear(input)
+        if count is not None:
+            return {
+                "logit": logit,
+                "loss": self.cross_entropy_reg(logit, count)
+            }
+        return {"logit": logit}
+
+    def cross_entropy_reg(self, logit, count):
+        if self.reg_mode == "l2":
+            reg_term = (self.linear.weight ** 2).sum()
+        elif self.reg_mode == "l1":
+            reg_term = abs(self.linear.weight).sum()
+        return -(F.log_softmax(logit, dim=1) * F.normalize(count.to(torch.float32), p=1.0, dim=1)).sum() + logit.shape[0] * self.reg_const * reg_term

del_model/runs/Nov20_11-56-17_ljw-System-Product-Name/events.out.tfevents.1732076861.ljw-System-Product-Name.1244212.2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:267073baf0ab3654eb653ae0648b751d92762e5c9b8db3b68dec2ae3be35cc1f
+size 19402

del_model/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f4cd672b6bea1c4f65e4fd1eceb6ac99713999f19259ba468c7b15d2398a7d06
+size 5304